一.安裝cuda11.0

cuda下載
此處的cuda下載請根據自己的系統資訊下載，否則使用會有問題。

安裝說明 :

    wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
    sudo sh cuda_11.0.3_450.51.06_linux.run

1.啟動安裝程式，一直按空格到最後，輸入accept接受條款

2.不安裝nvidia影象驅動，之前已經安裝過了

3.安裝完成後會有安裝的日誌記錄輸出， cat /var/log/cuda-installer.log,檢視無報錯一般就是安裝成功了，

4.配置環境變數

vim ~/.bashrc

加入路徑：

    export PATH=/usr/local/cuda/bin/:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64/

重新整理配置：source ~/.bashrc

5.cuda驗證

    nvcc -V

5.測試

首先安裝g++

    apt-get install g++

    #編譯並測試裝置 deviceQuery：
    cd /usr/local/cuda-11.0/samples/1_Utilities/deviceQuery
    sudo make
    ./deviceQuery
     
    #編譯並測試頻寬 bandwidthTest：
    cd ../bandwidthTest
    sudo make
    ./bandwidthTest

 

如果這兩個測試的最後結果都是Result = PASS，說明CUDA安裝成功啦

二.安裝cuDNN

下載cudnn

查詢nvcc -V對應的版本下載

1.安裝

解壓到cuda資料夾內 執行以下命令，進行安裝

    cd cuda/include/ 
    sudo cp ./include/cudnn.h /usr/local/cuda/include #執行即可，不需要做更改
    sudo cp ./lib64/libcudnn* /usr/local/cuda/lib64
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
 

2.建立軟連線

可以到下/usr/local/cuda/lib64/看一下自己的檔案

    cd /usr/local/cuda/lib64/ 
    sudo rm -rf libcudnn.so libcudnn.so.8
    sudo ln -s libcudnn.so.8.0.5 libcudnn.so.8
    sudo ln -s libcudnn.so.8 libcudnn.so

儲存後，建立連結檔案

    sudo vim /etc/ld.so.conf.d/cuda.conf 

按a進入插入模式，增加下面一行

    /usr/local/cuda/lib64 

按esc退出插入模式，按:wq儲存退出

最後在終端輸入

    sudo ldconfig 

後續Anaconda3等的安裝請參考此處：https://www.cnblogs.com/KdeS/p/14741567.html

文章參考地址：https://blog.csdn.net/pursuit_zhangyu/article/details/79362128