一 安裝顯卡驅動:
直接在系統軟件更新中選擇安裝:
或者選擇PPA源安裝,參照:https://blog.csdn.net/new_delete_/article/details/81544438
輸入命令測試是否安裝成功:
nvidia-smi
二、安裝CUDA
1、安裝gcc-6 g++-6(由於CUDA 9.0僅支持GCC 6.0及以下版本,而Ubuntu 18.04預裝GCC版本爲7.3)
sudo apt-get install gcc-6 g++-6
2、修改gcc、g++ 默認版本
cd /usr/bin
sudo rm gcc
sudo ln -s gcc-6 gcc
sudo rm g++
sudo ln -s g++-6 g++
3 去官網下載CUDA9.0 地址:https://developer.nvidia.com/cuda-toolkit-archive
3、安裝:
sudo chmod +x cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run
如果已經安裝驅動,就不要再選擇安裝驅動了
4、添加環境變量:
打開~/.bashrc
sudo gedit ~/.bashrc
在文件末尾添加以下命令:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
5、測試是否安裝成功:
nvcc -V
三、 安裝CUDNN
在官網下載相應的版本(deb包):地址:https://developer.nvidia.com/rdp/cudnn-download(需要註冊登陸)
2、安裝cudnn的deb包
sudo dpkg -i libcudnn7_7.4.1.5-1+cuda9.0_amd64.deb
完成