基础:
OS:ubuntu16.04-64bit (linux 4.13.0-36)
GPU: gtx-950
nvidia-driver version :384.111
cuda 9.0
cudnn-9.0-v7.1
相关链接及文件名:
nvidia-driver: https://www.geforce.cn/drivers 选择合适自己版本的
cuda 9.0 : https://developer.nvidia.com/cuda-downloads
cudnn-9.0 : https://developer.nvidia.com/cudnn
如何判断该安装什么版本的nvidia-driver,cuda,cudnn?
答:在系统中先查看本机的gpu是哪种型号的,可以用lspci |grep VGA 查看,然后用ubuntu-drivers devices 查看出驱动版本,一般在“driver:”那一行会有一个recommended 的版本号,可以去下载。然后选择相应的cuda,最后确定cudnn
依赖关系:driver->cuda->cudnn
安装步骤:
S1.安装好指定系统后,安装64bit 兼容IA32库:
根据该步骤:https://www.91linux.org/89.html
1、首先判断在用的架构
dpkg --print-architecture
输出当前的使用的架构,比如amd64
2、添加架构
dpkg --add-architecture i386
添加i386架构,然后刷新下源列表
sudo apt-get update
这时就已经包含i386的软件列表
3、安装i386
apt-get install iceweasel:i386
S2.禁用当前的nouveau 驱动,也即关闭Nouveau:
1)把驱动加入黑名单中: /etc/modprobe.d/blacklist.conf 在后面加入:
blacklist nouveau
2) 使用 dracut重新建立 initramfs image file :
* 备份 the initramfs file
$ sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
* 重新建立 the initramfs file
$ sudo dracut -v /boot/initramfs-$(uname -r).img $(uname -r)
3) 重启系统至文本模式,init 3 这个可以修改/etc/inittab 文件 init 3是文本模式。这时候重启就可以看到桌面非常的丑,就说明nouveau已经没有被加载了
4)检查nouveau driver确保没有被加载!
$ lsmod | grep nouveau
S3.安装nvidia-driver
1.按住CTRL+ALT+F1 进入命令行
2. sudo service lightdm stop 或者 sudo stop lightdm
3. sudo init 3
4. 安装驱动程序: #: sudo ./NVIDIA-Linux-x86_64-384.111.run //当前目录下执行NVIDIA驱动程序
5. 按照提示安装完成,简单方法重启就好了 sudo reboot
S4.安装cuda
sudo ./cuda_9.0176_384.81_linux.run
安装过程中选择询问是否需要安装nvidia-driver驱动时可以选择no ,因为上一步已经安装成功了。
成功后设置环境变量
export PATH="$PATH:/usr/local/cuda-9.0/bin"
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64"
运行nvidia-smi可以检查有没有安装成功
S5.安装cudnn
tar xvzf cudnn-9.0-linux-x64-v5.1.tgz
然后将库和头文件copy到cuda目录(一定是你自己安装的目录如/usr/local/cuda-9.0),
不过正确安装的话,ubuntu一般就会有软链接/usr/local/cuda -> /usr/local/cuda-9.0/
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*