2021-05-12 14:32:11
深度學習環境設定Ubuntu16.04+CUDA8.0+CUDNN5
深度學習從12年開始打響,設定深度學習環境軟體一直是一個頭疼的問題,如何安裝顯示卡驅動,如何安裝CUDA,如何安裝CUDNN;Ubuntu官方一直吐槽Nvidia顯示卡驅動有問題,網上大神也給出了關閉lightm巴拉巴拉之類的解決方法,有時候倒騰一星期也不見得安裝成功,之前聽一位師兄提起一種方法,然而在網上搜尋也搜尋不到,這裡給出安裝的教學,供大家參考
Ubuntu安裝好,顯示卡插上去之後,一般的,設定pytorch,tensorflow,mxnet,caffe等深度學習環境需要做三件事情
1.安裝NVIDIA驅動
通過命令列安裝,sudo apt-get install nvidia-375
2. 安裝cuda toolkit
sudo sh cuda_8.0.61_375.26_linux.r7un.26_linux-run
因為在第一步已經安裝NVIDIA驅動,所以在安裝cuda的時候不用安裝opengl,除此之外一路綠燈
3.安裝cudnn
3.1 進入下載的cudnn/include, cd cudnn/include, 拷貝cudnn.h 到/user/local/cuda-8.0/include/
sudo cp cudnn.h /user/local/cuda-8.0/include/
3.2 進入lib64資料夾,拷貝lib_cudnn_static.a 到/user/local/cuda-8.0/lib64資料夾下,sudo cp lib_cudnn_static.a /user/local/cuda-8.0/lib64/
3.3 拷貝libcudnn.so.5.1.3到/user/local/cuda-8.0/lib64/下,sudo cp libcudnn.so.5.1.3 /user/local/cuda-8.0/lib64/
3.4 進入/user/local/cuda-8.0/lib64/下並且建立軟連結,sudo ln -s libcudnn.so.5.1.3 libcudnn.so.5, sudo ln -s libcudnn.so.5 libcudnn.so
至此完畢
下面是之前的記錄
本文永久更新連結地址:http://www.linuxidc.com/Linux/2017-09/147180.htm
相關文章