ubuntu16.04安装英伟达驱动,cuda8.0和cudnn

禁用nouveal自带驱动(可选,推荐)

  控制台输入命令,创建一个文件通过命令

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

并添加如下内容:

blacklist nouveau

options nouveau modeset=0

更新一下:

sudo update-initramfs -u

有些博客上说要重启,此处建议不要重启,避免出现不能进入系统的问题.

安装GPU驱动

  Ctrl+Alt+F1进入控制台,输入以下命令:

sudo service lightdm stop

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get updates

sudo apt-get install nvidia-384

sudo apt-get install mesa-common-dev

sudo apt-get install freeglut3-dev

sudo reboot

其中 ‘nvidia-384’ 是驱动版本,上官网查询对应GPU型号的驱动即可。通过 ‘nvidia-smi’ 命令可以查看驱动是否正确安装。重启后可以正常使用图形界面。如果重启后分辨率有问题,请参考这里

CUDA安装

首先安装缺少的依赖包(强烈推荐,避免之后的安装出问题)

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libgl1-mesa-dev libglu1-mesa libglu1-mesa-dev libxi-dev

安装执行文件

  从英伟达驱动官网下载对应的cuda驱动程序,如cuda8.0,选择 ‘run’ 可执行文件下载。下载完成后输入命令安装:

sudo sh cuda8.0***.run

首先是大段大段的协议,一直按 NETER 即可,注意 “more(%)”。选择 “accept“ 接受,接下来根据提示安装。==注意:当询问是否安装 cuda toolkit, cuda-exampls 时选择yes, 询问是否安装其他驱动时一定要选择no。==其余如安装路径等默认即可

  设置环境变量,输入命令:

sudo gedit ~/.bashrc

export PATH=/usr/local/cuda-8.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

export CUDA_HOME=/usr/local/cuda

sudo source ~/.bashrc

sudo ldconfig

  检测cuda是否配置正确。输入命令:

‘nvcc - -version’ ,如果出现cuda版本信息等则说明安装正确。

还可以继续测试cuda的samples.参考这里

安装cudnn

  首先到官网下载对应版本的cudnn。安装cudnn比较简单,简单地说,就是复制几个文件:库文件和头文件。将cudnn的头文件复制到cuda安装路径的include路径下,将cudnn的库文件复制到cuda安装路径的lib64路径下。具体操作如下:

tar -zxvf cudnn-8.0-linux-x64-v7.tgz

cd cuda (就是刚刚解压出来的文件夹)

sudo cp include/cudnn.h /usr/local/cuda/include/

sudo cp lib64/* /usr/local/cuda/lib64/

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.7 (#删除原有动态文件,版本号注意变化,可在cudnn的lib64文件夹中查看)

sudo ln -s libcudnn.so.7.0.2 libcudnn.so.7 (生成软衔接(注意这里要和自己下载的cudnn版本对应,可以在/usr/local/cuda/lib64下查看自己libcudnn的版本))

sudo ln -s libcudnn.so.7 libcudnn.so

sudo ldconfig -v (立刻生效)

  查看安装cudnn后cuda是否依旧可用:

nvcc --version

这样,cudnn就安装完成了。如果要进一步验证,请参考这里

参考链接

  和前面的有重复:

1

2

3

你可能感兴趣的:(深度学习,linux)