首先去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_
沒有留言:
張貼留言