随着pytorch版本的更新,pytorch1.2.0官方编译的库最低只支持cuda9.2。所以这里更新一下软件驱动
1. 软件下载
所有的软件都要去nvidia官方下载,网速太慢,而且有时还断,还是很麻烦的。
百度云链接 链接: https://pan.baidu.com/s/1ted75tzap7FC_ysWFE0paQ 提取码: xwp9 复制这段内容后打开百度网盘手机App,操作更方便哦)
2. 安装
2.1 cuda9.2安装
cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64_(3).deb中其实已经有nvidia驱动。应该是nvidia-396,所以可以直接先安装cuda,安装完成之后用nvidia-smi
命令检查驱动是否已经安装成功。
cuda安装的命令:sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
如果安装cuda之后,用nvidia-smi
检查驱动已经成功安装则下一步就可以跳过
2.2 nvidia驱动安装
主要是一下命令:
chmod +x NVIDIA-Linux-x86_64-384.130.run
sudo ./NVIDIA-Linux-x86_64-384.130.run
可能遇到问题
- libkmod: ERROR …/libkmod/libkmod-config.c:635 kmod_config_parse: /etc/modprobe.d/blacklist.conf line 58: ignoring bad line starting with ‘option’
这可能是版本不匹配的问题,我就是换了版本,从410换成了418之后就解决了。
2.3 安装cudnn
具体可以参考我以前的博客。