2017年5月4日 星期四

Ubuntu 16.04上安裝CUDA-8.0

前一篇《Ubuntu 16.04只用內顯而不用獨顯》裝好了nvidia顯卡驅動程式,然後設定顯示用內部顯示而不是nvidia顯卡,這一篇則是繼續安裝CUDA-8.0

首先去CUDA下載頁面,找到所需的deb套件包,然後在virtual console內依序執行:
  1. sudo service lightdm stop
  2. sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
  3. sudo apt-get update
  4. sudo apt-get install cuda
  5. sudo service lightdm start
注意第2步的套件包名稱請修改成自己下載的那個。
下載完成以後,新增一個 /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
上面這個檔案是要讓所有的使用者都可以使用CUDA所需要預先設定的一些環境與路徑等等。如果只有某些使用者需要,那可以把這三行加在 ~/.bashrc 這個檔案裡面。
原本只要兩行,但發現執行 nvidia-smi 的時候會出現 "NVIDIA-SMI couldn't find libnvidia-ml.so library" 這種錯誤訊息,需要第三行才能夠正確的執行程式


參考資料:


_EOF_

沒有留言:

張貼留言