1、根據自己的系統在官網下載cuda (選擇runfile(local))
https://developer.nvidia.com/cuda-downloads
2、進入下載目錄,並執行
sh cuda_XXXX_linux.run
注意:1.在協議中選擇同意EULA(accept)
2.不安裝driver installation (no)
3.選擇cuda安裝目錄,如/home/yourname/cuda,以及cudasamples的目錄
3、下載cudnn,下載地址 https://developer.nvidia.com/cudnn
4、解壓cudnn
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
5、拷貝cudnn部分文件到cuda
cp cuda/include/cudnn.h cuda9/include/ cp cuda /lib64/libcudnn* cuda9/lib64 chmod a+r cuda/include/cudnn/h cuda/lib64/libcudnn*
6、修改個人用戶的環境變量
環境變量文件 ~/.bashrc位於home/yourname/~/.bashrc(用vim ~/.bashrc編輯)
“export PATH=$HOME/cuda9/bin:$PATH” “export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda9/lib64/”
修改之后 source ~/.bashrc使環境變量生效