如何在Ubuntu18.04下安装CUDA10.1和cudnn
CUDA10.1的版本比較新了,網上的一些教程不再適用。還有網上的部分介紹過于復雜,這里結合自己實際安裝經驗。介紹一個我的實現思路。
額外說一句gcc的事情,現在的版本gcc到了7.4,安裝CUDA10.1的時候,不再需要把gcc版本降低了。
安裝顯卡驅動
Ubuntu系統安裝好之后,原先自帶的顯卡驅動不是NVDIA的,需要卸載從新安裝。網上有說需要按 ctrl+alt+F1 進入tty,并且關閉圖形界面啥的。但我實際測試看來,不需要這樣操作也可以。而且進入tty,關閉界面著實復雜。有可能還操作不好。可以直接按我說的操作就可以,不需要進入tty。
首先,你要從NVIDA官網上下載好對應的顯卡驅動,作為準備。之后進入Ubuntu系統。啟動命令行。
填寫禁止配置的內容:
blacklist nouveau options nouveau modeset=0更新配置文件,并重啟電腦。
sudo update-initramfs -u sudo reboot切換到文件對應目錄,假設下載好的驅動文件是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,就裝一下
安裝CUDA
現在NVIDIA官網提供在線下載的方式了。假設你下載好了CUDA為xx.run文件。cudnn選擇好對應的版本直接就選Linux形式的就行了。下載好了是一個壓縮包。解壓之后里面有一個cuda文件夾。做好這樣的準備工作,下面來進行安裝。安裝過程參考自[2]
sudo sh xx.run安裝過程要特別注意,不要裝Driver,這里的Driver項是不選的。
安裝好之后,大概樣子應該是這樣的(這是我找的別人的圖,看到Summary信息比較對應就好了):
之后設置環境變量。
輸入 sudo gedit ~/.bashrc 打開主目錄下的 .bashrc文件添加如下路徑:
終端運行: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文件夾中為何不能粘貼圖片?
總結
以上是生活随笔為你收集整理的如何在Ubuntu18.04下安装CUDA10.1和cudnn的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 讲解泊松融合文章推荐
- 下一篇: 如何利用CycleGAN实现男女性别转换