Ubuntu18.04安裝安裝cuda和cudnn


一.安裝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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM