先安装显卡驱动
输入ubuntu-drivers devices
表示GPU是GeForce 940MX
系统推荐我用nvidia-driver-450-server来安装(并不知道这是什么)
同意他的建议就,sudo ubuntu-drivers autoinstall
安装后重启。
nvidia-smi 验证是否成功。
显示最高可安装的cuda版本为11.0,驱动器为450.
安装CUDA10.0
去官网下载cuda安装包(因为我打算装一个较低版本的pytorch,所以尽可能选择低版本的cuda安装,18.04对应的最低版本就是10.0了)
sudo sh 文件名
进行如下选择
出现问题Missing recommended library: libGLU.so,libX11.so,libXi.so,libXmu.so
这篇博客
卸载cuda
cd /usr/local/cuda/bin
sudo ./uninstall_cuda_9.0.pl
sudo rm -rf cuda-9.0
重装一遍就没啥问题了
nvcc -V
判断是否安装成功
安装CudNN 7.6.4
去官网下载安装包,需要登录。我发誓我真的很想登录,然后登录的链接极不稳定,试了快半个小时都没登上去。复制选项对应的链接去迅雷下载。
tar -xvf cudnn-10.0-linux-x64-v7.6.4.38.tgz
解压文件夹,执行下面的命令。可以把解压出来的包删掉了。
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
用上述代码判断是否安装成功。
结束。