更新NVIDIA显卡驱动、CUDA和cuDNN的步骤如下:
- 更新NVIDIA驱动:
# 首先移除旧的驱动
sudo yum remove nvidia-*
# 关闭图形界面
sudo systemctl isolate multi-user.target
# 禁用nouveau驱动
echo "blacklist nouveau" | sudo tee /etc/modprobe.d/blacklist-nouveau.conf
sudo modprobe -r nouveau
# 重新启动并进入文本模式
sudo reboot
# 下载合适的NVIDIA驱动(需要知道显卡型号和需要安装的驱动版本)
# 通过官网或者NVIDIA提供的命令行工具(如nvidia-smi)获取下载链接
# 安装驱动
sudo sh NVIDIA-Linux-x86_64-<VERSION>.run
# 重新启动
sudo reboot
- 更新CUDA Toolkit:
# 移除旧的CUDA版本
sudo yum remove cuda*
# 下载新的CUDA Toolkit(需要知道版本号和操作系统版本)
# 通常从NVIDIA官网下载.run文件
# 安装CUDA Toolkit
sudo sh cuda_<VERSION>_linux.run
# 配置环境变量
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
- 更新cuDNN:
# 首先下载cuDNN库(需要登录NVIDIA账户)
# 通常是一个.tgz文件
# 移动到cuDNN目录并解压
tar -xzvf cudnn-<VERSION>-linux-x64-v<VERSION>.tgz
# 拷贝库文件到CUDA目录
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
注意:在更新驱动和CUDA之前,请确保备份重要数据,并确认所有的软件依赖和环境变量设置。更新过程中可能需要关闭图形界面,并在文本模式下进行操作。在更新CUDA时,请使用与NVIDIA驱动兼容的版本。更新cuDNN时,请确保版本与CUDA Toolkit兼容。