前言
首先nvidia對於如何安裝cuda有線上說明書可參考,裡面很詳細:線上說明書
速成安裝紀錄
以下步驟幾乎都沒遇到問題,如果有問題的話可能要自己再回去說明書找,說明書很清楚記載每個步驟有問題要怎麼解
然後開始裝cuda前記得先上官網裝好你自己顯卡型號的driver(如果你是新卡或升級的話)
沒有driver可能會影響到雙螢幕、螢幕解析度不能調、裝cuda裝的不順等等的問題
官方載點: https://www.nvidia.com.tw/Download/index.aspx
I.清理環境
先清乾淨之前殘留的,不小心亂載到的檔案
sudo apt-get --purge remove nvidia-*
sudo rm -r /usr/local/cuda
sudo rm -r /usr/local/cuda-10.1
II.環境確定
確定好自己的環境
Verify You Have a CUDA-Capable GPU
lspci | grep -i nvidia
Verify You Have a Supported Version of Linux
uname -m && cat /etc/*release
Verify the System Has gcc Installed
gcc --version
Verify the System has the Correct Kernel Headers and Development Packages Installed
uname -r
III.按照官網的導引選cuda安裝方式
以下連結是我選好後的
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=debnetwork
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
IV.設定環境變數
這步漏掉的話,之後測試nvcc會直接找不到喔
打開bashrc文件
vim ~/.bashrc
在最後一行加入
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
V.重新開機
載完cuda的時候就會建議reboot了,所以就reboot一下
VI.完成!測試
基本上這樣就裝好了,再透過以下三種方法看一下是不是真的可以用了(即使前面中途沒error也有可能會有問題
測試
nvcc --version
測試
nvidia-smi
測試
python>>> import torch >>> torch.cuda.is_available()
這個可能要先自己裝一下pytorch
0 comments:
張貼留言
留言