在Ubuntu 18.04上安装CUDA 11和CuDNN 8时,如何避免驱动重复安装并正确配置环境变量?
时间: 2024-11-05 17:19:39 浏览: 25
为避免在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)
1. **检查并卸载现有的NVIDIA驱动**:
- 在终端输入`nvidia-smi`检查当前系统是否已安装NVIDIA驱动。
- 如果存在NVIDIA驱动,使用`sudo apt-get remove nvidia*`卸载当前驱动。
2. **安装NVIDIA显卡驱动**:
- 更新软件包列表:`sudo apt-get update`。
- 安装NVIDIA驱动:`sudo apt-get install nvidia-driver-版本号`,其中版本号为当前系统推荐的版本。
3. **安装CUDA**:
- 访问NVIDIA官方网站,下载CUDA Toolkit 11对应的.run文件。
- 在终端中,切换到下载目录,执行`sudo sh cuda_版本号Linux.run`。
- 在安装过程中,确保取消勾选安装NVIDIA驱动的选项。
4. **配置CUDA环境变量**:
- 确认CUDA安装路径,通常为`/usr/local/cuda-版本号`。
- 打开`~/.bashrc`文件:`vim ~/.bashrc`。
- 在文件末尾添加以下内容:
```
export PATH=/usr/local/cuda-版本号/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-版本号/lib64:$LD_LIBRARY_PATH
```
- 更新环境变量:`source ~/.bashrc`。
- 验证CUDA环境变量配置成功,使用`nvcc -V`查看CUDA编译器版本。
5. **安装CuDNN**:
- 下载与CUDA 11.0对应的CuDNN 8.05,解压后获取cuDNN文件夹。
- 将CuDNN的头文件复制到CUDA的include目录,库文件复制到lib64目录:
```
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
```
- 更新库文件缓存:`sudo ldconfig`。
- 验证CuDNN是否安装成功,可以通过检查`/usr/local/cuda/include`和`/usr/local/cuda/lib64`目录下的文件来确认。
以上步骤详细指导了如何在Ubuntu 18.04上安装CUDA和CuDNN,同时避免了驱动重复安装的问题。为了深入理解整个安装过程和各个步骤,推荐阅读《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)
阅读全文