Ubuntu16.04+cuda8.0+cudnn5.1安裝

1、安裝nvidia驅動

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 使環境變量生效。


3、測試cuda的Samples

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*

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章