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