安裝 CUDA
1. 下載cuda,連結在此,我的選擇如下圖,安裝的是cuda8版本,跳出來的選項兩個都要下載:
2. 在terminal下指令安裝cuda:
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.debsudo apt update && sudo apt install cuda -ysudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
3. 用以下指令編輯環境變數~/.bashrc,用vim或其他編輯方法都可以:
4. 在最後面加上下面cuda的環境程式碼,儲存後用source ~/.bashrc 更新環境變數:
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
5. 接下來測試看看cuda是否有安裝成功,使用cuda sample測試,將測試資料夾複製一份到當下路徑:
cp -r /usr/local/cuda/samples ./cuda_samplescd cuda_samplesmake -j8
如果你的使用者不是root會發生沒有權限的問題,可以在語法前加上sudo或是以下方法更改資料讀取權限,777是讀寫權限全開的意思,-R則是套用到底下所有的資料夾:
sudo chmod -R 777 cuda_samples/
6. 上述步驟好了後使用下面語法測試,最後出現pass就是成功了:
./bin/x86_64/linux/release/deviceQuery
安裝 cuDNN
1. 到這個nvidia官網下載cudnn關檔案,要有會員才能下載,不過有支援google帳號註冊方便很多,配合剛才安裝的cuda版本選擇對應的cudnn,進去下載頁面後載這四個檔案: