在Ubuntu系统中如何正确禁用NVIDIA Nouveau驱动,并安装CUDA、CUDNN及PaddlePaddle?请提供详细步骤。
时间: 2024-11-14 14:33:51 浏览: 4
在Ubuntu系统中安装CUDA、CUDNN以及PaddlePaddle之前,首先需要禁用可能会引起冲突的NVIDIA Nouveau驱动。这一步骤对于保证后续安装的顺畅和正确性至关重要。具体操作如下:
参考资源链接:[Ubuntu上CUDA与CUDNN的安装与卸载指南](https://wenku.csdn.net/doc/7myydppqat?spm=1055.2569.3001.10343)
1. 打开终端,使用文本编辑器(如nano或vim)编辑`/etc/modprobe.d/blacklist.conf`文件:`sudo nano /etc/modprobe.d/blacklist.conf`。
2. 在文件中添加以下两行内容:
```
blacklist nouveau
options nouveau modeset=0
```
3. 保存并关闭文件,执行`sudo update-initramfs -u`更新内核模块。
4. 重启系统,检查是否已成功禁用Nouveau驱动。可以通过运行`lsmod | grep nouveau`命令,如果没有输出则表示禁用成功。
之后,可以继续安装适合的NVIDIA显卡驱动。可以从NVIDIA官方网站下载对应的驱动文件,例如`NVIDIA-Linux-x86_64-410.93.run`。
安装驱动时,需要先停止X-Window服务并设置运行级别,执行以下命令:
```
sudo systemctl set-default multi-user.target
sudo systemctl stop gdm3
```
然后执行下载的驱动安装文件,并遵循提示完成安装。安装后,通过运行`sudo reboot`重启系统,以完成驱动的安装和配置。
接下来安装CUDA和CUDNN。首先确保系统中安装了必要的依赖库,然后根据需要下载CUDA和CUDNN的安装包。通常CUDA可以从NVIDIA的官方网站下载,而CUDNN则需要从NVIDIA开发者网站获取。
安装CUDA Toolkit,解压下载的文件,进入解压目录后运行安装脚本:
```
sudo sh cuda_10.0.130_410.48_linux.run
```
安装过程中可以选择安装CUDA驱动,但通常建议选择不安装(因为已安装了兼容的NVIDIA驱动),安装CUDA Toolkit和库即可。安装完成后,需要将CUDA的bin和lib目录添加到环境变量中,例如在`~/.bashrc`中添加:
```
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
同样,安装CUDNN后,根据CUDNN安装指南,将相关文件链接到CUDA目录。最后,安装PaddlePaddle。根据官方文档,可以通过以下命令安装:
```
pip install paddlepaddle-gpu==1.6.2.post107 -f ***
```
完成以上步骤后,你的Ubuntu系统应该已经正确配置了NVIDIA驱动、CUDA、CUDNN以及PaddlePaddle,可以开始进行深度学习的开发和训练工作了。
如果需要进一步了解和学习关于CUDA、CUDNN的安装细节以及卸载过程,你可以参考这份资料:《Ubuntu上CUDA与CUDNN的安装与卸载指南》。这份指南涵盖了详细的步骤和操作技巧,有助于你在处理深度学习框架及其依赖时,更加得心应手。
参考资源链接:[Ubuntu上CUDA与CUDNN的安装与卸载指南](https://wenku.csdn.net/doc/7myydppqat?spm=1055.2569.3001.10343)
阅读全文