1. 安装显卡驱动
ubuntu-drivers devices #可以显示推荐的显卡驱动版本
sudo ubuntu-drivers autoinstall #自动安装驱动
sudo reboot #重启之后可以通过“nvidia-smi”查看到显卡信息
可以看到显卡信息说明驱动安装成功
2. 安装cuda10.0
进入官网选择Archive of Previous CUDA Releases
这里要注意的是要是选择deb(network)的话在更新sudo apt update的时候会出问题,好像是因为国内网络的问题。所以建议选择runfile(local)
安装脚本运行过程中询问安装显卡驱动时输入否 同时安装toolkit的时候需要6个G的空闲磁盘空间,由于我的不够,所以安装到了/home/jerry/worspaces/cuda-10.0
安装脚本运行完之后要添加环境变量(根据自己的安装路径更改最后一条,/usr/local/cuda是安装过程中建立的软链接,不用改)
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-版本号/lib64
测试是否安装成功(sample的路径也是在安装脚本运行中自己设置的)
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
3. 安装cudnn7.3.0
在官网注册之后下载安装包
下载runtime,developer,samples
参开官方教程(根据自己下载的版本更改下列命令中的版本号)
cd ~/Downloads
sudo dpkg -i libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.4.2.24-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.4.2.24-1+cuda10.0_amd64.deb
安装完之后 测试:
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN
测试过程中会出现error while loading shared libraries: libcudart.so.10.0: cannot open shared object file: No such file or directory
解决办法(根据自己的安装路径改下列命令中的路径)
sudo cp /usr/local/cuda-10.0/lib64/libcudart.so.10.0 /usr/local/lib/libcudart.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcublas.so.10.0 /usr/local/lib/libcublas.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcurand.so.10.0 /usr/local/lib/libcurand.so.10.0 && sudo ldconfig
然后就成功了。