文章目录
前言
使用ubuntu做深度学习训练,安装驱动时遇到了很多坑,仅记录防止以后犯错。
一、安装NVIDIA驱动
1.1.查看显卡支持的驱动版本
ubuntu系统下:
打开终端 (ctrl + alt + T)输入:
lshw -numeric -C display
这一步主要确认硬件至此的驱动版本,如果知道显卡型号则可以跳过。
1.2.nvidia官网下载
下面的安装步骤是在ubuntu系统下的安装步骤,如果在windows下,在官网下载驱动后,next就可以了,十分方便,windows,yyds
1.2.1 在ubuntu下安装nvidia驱动可以选择从官网下载。nvidia官方驱动下载链接https://www.nvidia.com/Download/index.aspx
登录nvidia官网之后,在上图的界面选择对应的选项,点击搜索,下载,会下载一个nvidia-*.run文件。
1.2.2 禁用ubuntu自带的nouveau
在终端输入
sudo gedit /etc/modprobe.d/blacklist.conf
在文件后面添加
blacklist nouveau
options nouveau modeset=0
重启更新系统修改
sudo update-initramfs -u
sudo reboot
验证nouveau是否禁用
lsmod | grep nouveau
若成功禁用则在终端输入上述命令无任何反应。
1.2.3 开始安装
将下载的nvidia-*.run拷贝至/home文件夹内(这一步方便在安装时找到文件),按ctrl + alt + F1进入命令界面,提示输入密码,输入密码否登录。
然后依次执行如下命令
sudo service lightdm stop //关闭图形界面
sudo apt-get remove nvidia-* //卸载系统中存在的驱动
sudo chmod a+x NVIDIA-*.run //给文件权限
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau