這篇文章主要是講安裝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
-
安裝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自帶的驅動,很麻煩,搞不好需要重裝系統。
-
配置環境變量:
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,不用管。
-
重新啓動 :
sudo reboot
在命令行輸入nvcc -V來檢測一下CUDA是否安裝成功了吧!
安裝cudnn
- 解壓安裝包:在cudnn壓縮包所在位置打開終端,解壓cudnn-10.0-linux-x64-v7.tgz命令如下:
tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
執行命令後,你會發現多了一個cuda文件夾,如下圖所示,如果 解壓過程出錯,可以多試幾次,如果還不行,那就是壓縮包出問題了。
-
複製操作:
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的安裝命令的截圖如下所示:
到這一步,恭喜你安裝基本完成了!你可以用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