sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
執行完上述後,reboot重啓後執行:
nvidia-smi
如果出現了你的GPU列表,則說明驅動安裝成功了。另外也可以通過:
nvidia-settings查看自己機器上詳細的GPU信息,如下:
2、安裝CUDA
sudo sh cuda_8.0.44_linux.run
因爲驅動之前已經安裝,這裏就不要選擇安裝驅動。其餘的都直接默認或者選擇是即可。
修改/etc/profile,添加以下內容:export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH
執行 . /etc/profile 使環境變量生效。
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
./deviceQuery
如果顯示的是一些關於GPU的信息,則說明安裝成功了。
4、安裝cudnn5.1
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.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*