在Ubuntu 18.04上安装CUDA 11和CuDNN 8时,如何避免驱动重复安装并正确配置环境变量?
时间: 2024-11-05 17:19:40 浏览: 7
为了避免在安装CUDA和CuDNN时重复安装显卡驱动并正确配置环境变量,首先要确保不从CUDA安装程序中选择安装或更新NVIDIA驱动。具体步骤如下:
参考资源链接:[Ubuntu 18.04 安装CUDA 11 & CuDNN 8 教程](https://wenku.csdn.net/doc/3q6djacmgk?spm=1055.2569.3001.10343)
1. 更新系统软件包列表并安装NVIDIA显卡驱动,可以通过Ubuntu软件中心或者使用命令行运行`sudo ubuntu-drivers autoinstall`。
2. 安装CUDA时,从NVIDIA官网下载对应版本的.run文件。在安装过程中,使用`sudo ./cuda_11.0.3_450.51.06_linux.run`命令启动安装,当提示是否安装NVIDIA驱动时,选择“No”。
3. 完成CUDA安装后,配置环境变量以确保系统能够识别CUDA工具。编辑`.bashrc`文件,添加以下内容:
```
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
4. 保存并执行`source ~/.bashrc`使配置立即生效。
5. 为安装CuDNN,首先需要下载与CUDA 11.0版本对应的CuDNN压缩包,解压后将头文件、库文件和库文件软链接复制到CUDA安装目录。
```
sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/lib* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/lib*
```
6. 运行`sudo ldconfig`更新系统库文件路径。
7. 最后,验证安装是否成功,使用命令`nvcc -V`检查CUDA版本,以及使用`cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2`检查CuDNN版本。
通过这些步骤,可以确保在Ubuntu 18.04上安装CUDA 11和CuDNN 8时避免驱动重复安装并正确配置环境变量。如果你希望获取更详尽的指导和操作步骤,请参阅《Ubuntu 18.04 安装CUDA 11 & CuDNN 8 教程》。
参考资源链接:[Ubuntu 18.04 安装CUDA 11 & CuDNN 8 教程](https://wenku.csdn.net/doc/3q6djacmgk?spm=1055.2569.3001.10343)
阅读全文