CUDA 9.0在Ubuntu上的安裝
阿新 • • 發佈:2019-02-16
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