1. 程式人生 > >CUDA 9.0在Ubuntu上的安裝

CUDA 9.0在Ubuntu上的安裝

1.下載CUDA

在以下網址下載9.0的deb檔案(注意選取合適的平臺):

https://developer.nvidia.com/cuda-downloads

(如果當前頁面為9.0以上版本,可在必應上用“cuda 9.0 download ”作為關鍵詞檢索定位相關下載網頁)

2.終端安裝deb檔案

在終端執行如下命令:

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub  #執行完上一句指令後,末尾會有此指令的提示
sudo apt-get update
sudo apt-get install cuda

3.修改環境變數

在終端開啟並修改.bashrc檔案

sudo gedit ~/.bashrc

將如下內容新增到.bashrc檔案末尾:
export CUDA_HOME=/usr/local/cuda-9.0 
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64 
export  PATH=${CUDA_HOME}/bin:${PATH} 

修改完成後,在終端輸入指令,使環境變數立即生效:

sudo ldconfig 

4.驗證

在終端輸入命令,實時檢視GPU的使用情況:

watch -n 1 nvidia-smi

5.CUDA 9.0和CUDA 8.0共存

如果已經安裝CUDA 9.0,還需要安裝CUDA 8.0,並使預設版本為8.0,則下載deb安裝並執行如下類似指令:

sudo dpkg -i cuda8.deb
sudo apt-get update
sudo apt-get install cuda-8-0

並修改.bashrc檔案內容。

CUDA 8有個補丁檔案,最好安裝上:

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update  
sudo apt-get upgrade cuda-8-0