首先去CUDA下載頁面,找到所需的deb套件包,然後在virtual console內依序執行:
- sudo service lightdm stop
 - sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
 - sudo apt-get update
 - sudo apt-get install cuda
 - sudo service lightdm start
 
下載完成以後,新增一個 /etc/profile.d/cuda.sh 的檔案,加入下面三行:
- export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
 - export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 - export LD_PRELOAD=/usr/lib/nvidia-375/libnvidia-ml.so
 
原本只要兩行,但發現執行 nvidia-smi 的時候會出現 "NVIDIA-SMI couldn't find libnvidia-ml.so library" 這種錯誤訊息,需要第三行才能夠正確的執行程式
參考資料:
- INSTALLING CUDA 8 ON UBUNTU 16.04
 - 在 Ubuntu 16.04 安裝 CUDA 8.0 和 TensorFlow 0.11 並使用內顯顯示畫面
 - 深度学习主机环境配置: Ubuntu16.04+Nvidia GTX 1080+CUDA8.0
 
_EOF_
沒有留言:
張貼留言