一 安裝顯卡驅動:
直接在系統軟件更新中選擇安裝:
或者選擇PPA源安裝,參照:https://blog.csdn.net/new_delete_/article/details/81544438
輸入命令測試是否安裝成功:
nvidia-smi
二 安裝CUDA10
CUDA官網選擇適合自己系統的版本下載。
2、運行命令安裝CUDA10
sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
3、添加環境變量:
打開 .bashrc
sudo gedit ~/.bashrc
在末尾添加:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存退出
刷新環境變量:
source ~/.bashrc
測試CUDA是否安裝成功:
nvcc -V
界面如下:
三 安裝CUDNN
在官網下載安裝包,需要註冊登錄才能下載,選擇適合自己的版本(選擇deb安裝包,可直接安裝完成,而不需要人爲拷貝):
在這我選擇:cuDNN Runtime Library for Ubuntu18.04 (Deb)
安裝CUDNN:
sudo dpkg -i libcudnn7_7.4.1.5-1+cuda10.0_amd64.deb
完成