2018年5月31日 星期四

ubuntu 16.04 安裝 cuda 9.0

ubuntu 16.04 安裝 cuda 9.0

驅動程式可以不安裝過程會自己安裝,先安裝也會重新安裝cuda選用的版本,整個流程跑完要手動重新開機。
因為有安裝驅動,建議下面流程跑完就馬上重新開機,不然用一用可能會當機。安裝驅動之後重新開機第一次會比較久,等一下會黑畫面什麼都沒有很像死機了。
過程涉及安裝驅動~如果有發生什麼衝突可能導致系統崩潰無法開啟,這點要注意一下~

可以不用點有包含再底下指令(cuda版本有升級文章還沒更新就要手動改版本號)
安裝指令
# 加入驅動鏡像站
sudo add-apt-repository ppa:graphics-drivers/ppa -y

# 升級系統
sudo apt-get update
sudo apt-get upgrade -y

# 安裝 cuda
cd ~/Downloads
sudo wget -c https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda -y

# 加入函式庫
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
到這邊就裝好了,還要配置他的bin跟lib位置,否則直接打 nvcc 會出現找不到
方法是參考官方說明網站:docs.nvidia.com
# 函式庫標頭檔路徑
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}

# 函式庫路徑(這個好像不用)
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64 ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
這樣就配置好了可以打
nvcc --version
只不過剛剛的配置方法只有對這個終端機有用,換終端機或是重開就沒了
想要真正加入的話把那兩行寫入 vi .bashrc 這個文件裏面開機自動執行就好了
vi .bashrc
加入最尾端或是隨意找個地方寫也可以,然後重新登入再打測試版本的指令就有了。
最後測試一下到底有沒有裝對。
這邊記得先重開機一次否則測試結果會失敗歐
git clone https://github.com/hunandy14/cuda_copyArr
cd cuda_copyArr
make run
看一下有沒有跑出 test ok 有的話就是一切都正常了,包含驅動。(失敗的話重新開機再試試看)
這是一個簡單的cuda程式,其中我有寫一個比較容易的呼叫使用記憶體的函式,適合新手學習使用。
寫文章測試的時候 cuda9.0 選用的版本 384 而當前最新版本是 390 ,建議不要更動這個版本裝好就別升級了,可能會發生一些無法預測的錯誤。

沒有留言:

張貼留言