Ubuntu1804安裝cuda10和cudnn TensorRT

這篇文章主要是講安裝cuda和cudnn的,我會一步一步詳細說明。

Ubuntu安裝Nvidia顯卡驅動的操作,我這裏就省略了。。。如果不知道如何裝英偉達獨顯驅動,請評論區留言,留言超過十人,我會單獨寫一篇Ubuntu安裝Nvidia顯卡驅動的文章。


我的環境:Ubuntu1804    顯卡驅動:435.21   我要安裝cuda10  cudnn10.0

先分享一下我的資源

cuda10、cudnn10.0 鏈接: https://pan.baidu.com/s/1-ww32KqPhuOODiTeb-mgDA  密碼: p7ka

cuda的官方下載鏈接:https://developer.nvidia.com/cuda-toolkit-archive

cudnn的官方下載鏈接:https://developer.nvidia.com/rdp/cudnn-download

安裝CUDA

  1.  安裝cuda包:進入到你cuda10的安裝包,也就是cuda_10.0.130_410.48_linux.run的所在位置,運行如下命令:

    sudo sh cuda_9.0.176_384.81_linux.run

    然後一直回車回車,確定,直到出現是否安裝Nvidia驅動軟件(當然是英文的,什麼Nvidia Drivers。。。具體忘了) ,輸入N,爲啥,這裏默認你已經安裝過Nvidia驅動了。如果你還沒安裝Nvidia的驅動,那需要專門單獨先安裝好了再說。因爲要禁用Ubuntu自帶的驅動,很麻煩,搞不好需要重裝系統。

  2. 配置環境變量:

    sudo gedit  /etc/profile
    #在文件末尾中添加路徑,也就是安裝目錄
    export  PATH=/usr/local/cuda-10.0/bin:$PATH
    export  LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64$LD_LIBRARY_PATH

    這裏,執行完sudo gedit /etc/profile命令後,會進入profile編輯頁面,在文件最後添加另外兩行,保存就可以了。這裏可能會提示有WARNING,不用管。

  3. 重新啓動 :

    sudo reboot

    在命令行輸入nvcc -V來檢測一下CUDA是否安裝成功了吧!

安裝cudnn

  1.  解壓安裝包:在cudnn壓縮包所在位置打開終端,解壓cudnn-10.0-linux-x64-v7.tgz命令如下:
    tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

    執行命令後,你會發現多了一個cuda文件夾,如下圖所示,如果 解壓過程出錯,可以多試幾次,如果還不行,那就是壓縮包出問題了。

  2. 複製操作:

    sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

    整個cudnn的安裝命令的截圖如下所示: 

cudnn的安裝

到這一步,恭喜你安裝基本完成了!你可以用TensorFlow的GPU代碼來檢測一下。

import tensorflow as tf
print(tf.test.is_gpu_available())
#如果返回爲true則可以調用GPU

如果爲False,那你遇到的問題和我的一樣,我參考了這篇文檔完美解決:https://blog.csdn.net/u012388993/article/details/102573117這裏,先感謝一下那個博主。                                                                       

爲什麼返回爲False呢?主要是因爲環境變量沒配置好

1、用下邊代碼新建文件

sudo vi /etc/profile.d/cuda.sh

2、將下邊的內容粘貼進去並保存(不會vim的,輸入i進入插入模式,粘貼,esc, :wq! )

export PATH=$PATH:/usr/local/cuda/bin
export CUDADIR=/usr/local/cuda

3、再新建一個文件

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

4、同2

/usr/local/cuda/lib64

5、設置鏈接,最後sudo ldconfig出現某某文件not a symbolic link問題,但不用管,直接重啓一下電腦就好了

sudo ldconfig

OK,大功告成了 ,再用代碼測試一下吧,肯定返回True!

如果覺得有用的話,給我點個贊鼓勵一下吧。

安裝TensorRT

下篇文章我會寫安裝使用TensorRT加速推理的博客,如果感興趣,請關注我,我們一起成長!

補充一個安裝QQ、微信的鏈接:https://github.com/wszqkzqk/deepin-wine-ubuntu 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章