這邊建議裝 CUDA 8 就好了
2. 用以下指令安裝 CUDA
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt update && sudo apt install cuda -y
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/lib/nvidia-384
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:/usr/local/cuda/bin
cd
cp -r /usr/local/cuda/samples ./cuda_samples
cd cuda_samples
make -j8
編譯完成後執行以下指令,如果最後出現Pass,代表安裝成功
./bin/x86_64/linux/release/deviceQuery
下載cudnn,進入網站後點選 Download
再來會要你登入帳號,如果沒有帳號就去註冊一個吧。
根據你的作業系統以及cuda版本來選擇要下載的安裝檔,總共有兩個要下載
tar xvfz cudnn-8.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/* /usr/local/cuda/lib64
sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda8.0_amd64.deb
cp -r /usr/src/cudnn_samples_v7 ./
cd cudnn_samples_v7/mnistCUDNN/
make -j8
./mnist
sudo apt install -f
到此 CUDA 跟 cudnn 就安裝完成了