Linux 上的 NVIDIA 驅動

Nvidia driver on Ubuntu

find here

• 到 這裡 查看最新版本
  • nvidia release
  • 34x: 傳統架構版本
  • 387: 新版短期
  • 384: 新版長期
• 刪除舊版

sudo apt-get purge nvidia*

• 安裝新版

sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
sudo apt-get install nvidia-387
sudo reboot # 重開機

• 檢查

lsmod | grep nvidia

nvidia-smi

cuda

sudo apt install cuda-9
sudo apt install cuda-nvcc-9-1

• 手動安裝 到 這裡 下載 runfile(local)

sudo sh cuda_9.1.85_387.26_linux.run

• 檢查 注意：gcc 要是 < 6 的版本 @@

cp -r /usr/local/cuda-9.1/samples ./
cd samples
make

cudnn

到 這裡 下載

• cuda 的資料夾要在 /usr/local/cuda/，so 不管裝在哪裡，都給一個 link 到 /usr/local/cuda
• ubuntu 版:
  • 下載 libcudnn7…deb, libcudnn7-dev…deb, libcudnn7-doc…deb
  • 三個都做 sudo dpkg -i
• 其他
  • 下載壓縮檔
  • 解壓縮: tar -xzvf cudnn-9.0-linux-x64-v7.tgz
  • 安裝:

    $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
    $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*