如何在Ubuntu18.04下安裝CUDA10.1和cudnn

CUDA10.1的版本比較新了,網上的一些教程不再適用。還有網上的部分介紹過於複雜,這裏結合自己實際安裝經驗。介紹一個我的實現思路。

額外說一句gcc的事情,現在的版本gcc到了7.4,安裝CUDA10.1的時候,不再需要把gcc版本降低了。


安裝顯卡驅動

Ubuntu系統安裝好之後,原先自帶的顯卡驅動不是NVDIA的,需要卸載從新安裝。網上有說需要按 ctrl+alt+F1 進入tty,並且關閉圖形界面啥的。但我實際測試看來,不需要這樣操作也可以。而且進入tty,關閉界面着實複雜。有可能還操作不好。可以直接按我說的操作就可以,不需要進入tty。

首先,你要從NVIDA官網上下載好對應的顯卡驅動,作爲準備。之後進入Ubuntu系統。啓動命令行。

  1. 刪Ubuntu系統除舊驅動
# 刪除
sudo apt-get purge nvidia*
  1. 禁止自帶的nouveau nvidia驅動
# 打開配置文件
sudo vim /etc/modprobe.d/blacklist-nouveau.conf

填寫禁止配置的內容:

blacklist nouveau
options nouveau modeset=0

更新配置文件,並重啓電腦。

sudo update-initramfs -u
sudo reboot
  1. 安裝準備好的顯卡驅動文件

切換到文件對應目錄,假設下載好的驅動文件是xx.run

sudo sh xx.run

如果使用nvidia-smi命令可以出現類似這樣的畫面,就說明驅動安裝好了。
在這裏插入圖片描述

可能存在的問題

如果你在安裝過程中報錯
Unable to find the development tool cc in your path; please make sure that you have the…
可能是你的系統裏沒有裝gcc,那就裝上gcc[1],類似的,報找不到make,就裝一下

sudo apt-get install gcc
sudo apt-get install make

安裝CUDA

現在NVIDIA官網提供在線下載的方式了。假設你下載好了CUDA爲xx.run文件。cudnn選擇好對應的版本直接就選Linux形式的就行了。下載好了是一個壓縮包。解壓之後裏面有一個cuda文件夾。做好這樣的準備工作,下面來進行安裝。安裝過程參考自[2]

sudo sh xx.run

安裝過程要特別注意,不要裝Driver,這裏的Driver項是不選的。
在這裏插入圖片描述

安裝好之後,大概樣子應該是這樣的(這是我找的別人的圖,看到Summary信息比較對應就好了):
在這裏插入圖片描述

之後設置環境變量。
輸入 sudo gedit ~/.bashrc 打開主目錄下的 .bashrc文件添加如下路徑:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

終端運行:source ~/.bashrc 使之生效。

檢查安裝是否成功:

輸入 nvcc --version,如果顯示如下文字就說明安裝成功了:
在這裏插入圖片描述


安裝cudnn

把準備好的cudnn文件複製到已經安裝好的cuda文件夾裏。cuda安裝在usr文件夾裏。點開主目錄裏下面的計算機就可以找到。但是在這裏沒有辦法用拖拽的方式進行復制黏貼(因爲沒有超級權限),需要使用命令行的方式。安裝過程參考自[3]

首先進入cudnn對應的文件夾裏。之後敲如下命令

sudo cp cuda/include/* /usr/local/cuda/include
sudo cp cuda/lib64/*   /usr/local/cuda/lib64

複製成功就可以了。

最後額外說一點,如果你想卸載顯卡驅動的話,可以用這個命令來卸載:

sudo apt-get remove nvidia*

如果想查看隱藏文件夾,可以通過Ctrl + h組合鍵。

參考資料

[1]成功解決ERROR: Unable to find the development tool cc in your path; please make sure that you have the
[2]Ubuntu:安裝 cuda10.1 驅動
[3]LINUX安裝cuDNN
[4]ubuntu中的usr/share/backgrounds文件夾中爲何不能粘貼圖片?

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