ubuntu16配置Mask R-CNN深度学习环境——驱动显卡+cuda+pytorch

最新推荐文章于 2023-03-11 14:07:21 发布

原创

最新推荐文章于 2023-03-11 14:07:21 发布 · 1.2k 阅读

CC 4.0 BY-SA版权

文章标签：

本文详细介绍了在Ubuntu16.04上配置深度学习环境的过程，包括如何卸载和安装NVIDIA显卡驱动，安装CUDA8.0，以及如何避免安装过程中遇到的问题。同时，文章还提供了安装PyTorch的步骤，验证了CUDA和PyTorch的正确配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、下载安装nvidia显卡驱动
先去nvidia官网查看对应显卡适配驱动型号

先卸载原有N卡驱动
#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*
#for case2: original driver installed by runfile:
sudo chmod +x *.runsudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

如果原驱动是用apt-get安装的，就用第1种方法卸载。
如果原驱动是用runfile安装的，就用–uninstall命令卸载。其实，用runfile安装的时候也会卸载掉之前的驱动，所以不手动卸载亦可。

禁用nouveau驱动
sudo gedit /etc/modprobe.d/blacklist.conf
在文本最后添加：（禁用nouveau第三方驱动，之后也不需要改回来）
blacklist nouveau
options nouveau modeset=0
然后执行：
sudo update-initramfs -u
重启后，执行：lsmod | grep nouveau。如果没有屏幕输出，说明禁用nouveau成功。
禁用X-Window服务
sudo service lightdm stop #这会关闭图形界面，但不用紧张
按Ctrl-Alt+F1进入命令行界面，输入用户名和密码登录即可。
命令行安装驱动
#给驱动run文件赋予执行权限：sudo chmod +x NVIDIA-Linux-x86_64-390.87.run大小写键盘问题，Ubuntu不认小键盘，所以一定要关掉NumLock，使用字母键盘上面的数字按键进行输入。#后面的参数非常重要，不可省略：sudo ./NVIDIA-Linux-x86_64-390.87.run –no-opengl-files –no