【深度學習環境】Linux下安裝、切換cuda和cudnn

安裝cuda

1. 打開下載鏈接 ，點擊Download Now
2. 選擇操作系統、架構、版本、安裝方式
   ps：如果要選擇非最新版本cuda, 點擊下方的legacy releases
   
3. 輸入Installation Instructions中的指令獲取安裝包並開始安裝

### 獲取安裝包
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
### 開始安裝
sudo sh cuda_10.1.243_418.87.00_linux.run

4. 安裝過程

• 首先會出現一系列協議，一直按住Enter鍵。之後會出現一系列選項。
• 第一個選擇accept
• 第二個記得取消驅動選項.
• 後面全部選擇y
  
  

5. 設置cuda環境變量
   輸入 sudo gedit ~/.bashrc 打開主目錄下的.bashrc文件添加如下內容:

### 注意 10.1爲版本號，應當替換爲剛剛下載的版本
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

6. 終端運行：source ~/.bashrc 使之生效
7. 檢查是否安裝成功:
   輸入nvcc --version，如果顯示如下文字就說明安裝成功
   
8. 如果顯示錯誤，註銷一下用戶

安裝cudnn

1. 打開下載鏈接，選擇相應的版本並下載
2. 進入cudnn壓縮包所在的文件夾，解壓文件

tar -xvf cudnn-10.1-linux-x64-v7.6.4.38.tgz

3. 將文件拷貝到相應cuda目錄下

### 注意 10.1爲版本號，應當替換爲剛剛下載的版本
### 注意安裝過程中不要隨意切換路徑
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

4. 查看cudnn版本

### 注意 10.1爲版本號，應當替換爲剛剛下載的版本
cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2

出現如下的信息，則爲配置成功

切換cuda版本

如果有安裝多個cuda, 只需修改.bashrc文件中的版本號即可隨意切換cuda版本。具體方法見安裝cuda一節中的 5. 設置cuda環境變量

參考鏈接

1. https://blog.csdn.net/sss_369/article/details/94591280
2. https://blog.csdn.net/wanzhen4330/article/details/81699769
3. https://zhuanlan.zhihu.com/p/35834028