1.cuda 的安裝
cuda 下載地址 https://developer.nvidia.com/cuda-downloads
2. 安裝 cuda
以下步驟全都是 ROOT 權限操作
chmod +x cuda_10.1.168_418.67_linux.run 將.run 文件變成可執行文件
./ cuda_10.1.168_418.67_linux.run
運行安裝 cuda
選擇安裝 cuda，這裡我沒有選擇安裝顯卡驅動。有需要可以直接選擇
安裝完成
寫入環境變量
vim /etc/profile
加入 cuda 環境變量
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
保存退出後同步環境變量
source /etc/profile
nvcc -V
驗證 cuda 版本，有輸出 cuda 版本，cuda 安裝成功。
3.cudnn 的安裝
下載 cudnn https://developer.nvidia.com/rdp/cudnn-archive
下載完後解壓，進入解壓後的文件
創建 cuda 的軟連
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
驗證 cudnn7 安裝是否成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
驗證 cudnn8 安裝是否成功
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
CUDA+CUDNN 安裝完成