1 GPU驅(qū)動的安裝
直接使用ubuntu自帶的 軟件和更新 軟件來安裝就好,點(diǎn)附加驅(qū)動色乾,找到NVIDA
(或者安裝cuda時也可以安裝)
貌似目前conda在安裝pytorch時會自動幫你安裝cuda,cudnn,nvcc好像一般用不到柏副。所以不用單獨(dú)安裝他們,也就是說安裝完驅(qū)動,然后nvidia-smi查看最高支持的cuda版本就好了糕非。
2 cuda的安裝
首先查看當(dāng)前驅(qū)動可支持的CUDA版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
或者直接nvidia-smi查看最高支持的cuda版本。
然后下載相應(yīng)版本的CUDA
官網(wǎng)https://developer.nvidia.com/cuda-toolkit-archive
找到cuda10.0球榆,點(diǎn)進(jìn)去
這里選擇runfile方式安裝朽肥,deb安裝容易出錯
1)分別把第一個主文件和下面的補(bǔ)丁下載下來,在下載的文件夾處打開終端持钉,輸入以下命令進(jìn)行安裝(如果安裝失敗嘗試deb文件安裝的方法)
sudo sh cuda_10.0.130_410.48_linux.run
sudo sh cuda_10.0.130_410.48_linux.run為該文件名衡招,看情況為定
注:如果沒有g(shù)cc和g++則需要先安裝
sudo apt-get install gcc
sudo apt-get install g++
如果安裝CUDA 9.0則需要制定版本的gcc,g++。cuda9.0僅支持GCC 6.0及以下版本每强,而Ubuntu 18.04預(yù)裝GCC版本為7.3蚁吝,
故手動進(jìn)行降級:(忽略)
sudo apt-get install gcc-4.8
sudo apt-get install g++-4.8
接著按下面的步驟選擇:
一路回車
accept
n(不要安裝driver,已有驅(qū)動)
y
y
y
2)安裝完成后舀射,設(shè)置環(huán)境變量。
打開主目錄下的 .bashrc文件添加如下路徑怀伦,例如我的.bashrc文件在/home/wangyuanwei下脆烟,如果沒有找到,則按Ctrl+H鍵顯示隱藏文件房待。
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
終端運(yùn)行:source ~/.bashrc
檢查:nvcc --version邢羔,如果顯示下面的文子就說明安裝成功了。
3.cudnn的安裝
官網(wǎng)https://developer.nvidia.com/rdp/cudnn-archive
這里選cuDNN Library for Linux(Deb安裝容易出錯)
下載下來桑孩,解壓下載的文件拜鹤,可以看到cuda文件夾,在當(dāng)前目錄打開終端流椒,執(zhí)行如下命令:(也就是把下載的cudnn文件復(fù)制到相應(yīng)的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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
新版本cudnn加上敏簿,用于查版本信息
sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include/
或直接
sudo cp ./* /usr/local/cuda/lib64
安裝完成
3)查看cudnn版本
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
新版本為以下命令
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
如果出現(xiàn)下圖所示版本信息,說明安裝成功宣虾。
4.安裝NCCL
下載地址https://developer.nvidia.com/nccl/nccl-download
找到對應(yīng)版本下載惯裕,ubuntu可以直接安裝
卸載CUDA
卸載CUDA很簡單,一條命令就可以了绣硝,主要執(zhí)行的是CUDA自帶的卸載腳本蜻势,讀者要根據(jù)自己的cuda版本找到卸載腳本:
sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
或
Installation Guide Linux :: CUDA Toolkit Documentation (nvidia.com)
sudo apt-get --purge remove "*cublas*" "*cufft*" "*curand*" \
"*cusolver*" "*cusparse*" "*npp*" "*nvjpeg*" "cuda*" "nsight*"
卸載之后,還有一些殘留的文件夾鹉胖,之前安裝的是CUDA 10.0握玛」话可以一并刪除:
sudo rm -rf /usr/local/cuda-10.0/
Ubuntu18.04問題——N: 無法安全地用該源進(jìn)行更新,所以默認(rèn)禁用該源挠铲。
打開軟件軟件和更新(不是換源)
然后點(diǎn)擊其它軟件
里邊有一個網(wǎng)址與你報(bào)錯的網(wǎng)址一樣冕屯,而且是勾選狀態(tài)的
取消選擇,然后關(guān)閉
參考資料
https://blog.csdn.net/wanzhen4330/article/details/81699769
https://blog.csdn.net/u010801439/article/details/80483036
https://www.cnblogs.com/carle-09/p/12373013.html
(17條消息) Ubuntu18.04問題——N: 無法安全地用該源進(jìn)行更新市殷,所以默認(rèn)禁用該源愕撰。_qq494370的博客-CSDN博客