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

来源:互联网 发布:董易奇八字软件 编辑:程序博客网 时间:2024/05/17 08:38
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*

原创粉丝点击