1 系统相关配置准备
1.1 查看自己的电脑是否具有支持CUDA的GPU
~$ lspci | grep -i nvidia
说明我有一块可以用的Quadro M4000显卡,可以在https://developer.nvidia.com/cuda-gpus查看显卡信息GPU是否有CUDA-caplable
1.2 确认你的linux系统是否支持CUDA
查看本机系统信息
~$ uname -m && cat /etc/*release
CUDA开发工具只支持几个特定的LINUX发行版本
1.3查看系统是否安装了gcc
~$ gcc --version
1.4 验证系统是否有正确的kernel headers以及development packages
~$ uname -r
~$ sudo apt-get install linux-headers-4.4.0-21-generic
2.1 确认之前的准备工作已经完成
2.2 禁用Nouveau驱动器
方法一:
~$ lsmod | grep nouveau
在相关目录下创建一个黑名单文件
~$ sudo vim /etc/modprobe.d/blacklist-nouveau.conf
写入以下信息:
blacklist nouveau
options nouveau modeset=0
重新生成kernel initramfs
方法二:使用另一种方法,打开系统设置—>软件与更新—>附加驱动,修改至下图所示
然后执行重启
执行以下命令查看驱动是否安装成功
3 安装cuda
3.1 到官网https://developer.nvidia.com/cuda-toolkit-archive下载相应版本的cuda文件,
在文件所在目录(我的在下载目录)执行以下命令,
~/下载$ sudo sh cuda_9.0.176_384.81_linux.run
一直按enter,直到问是否安装OpenGL时选择no(如果电脑为双显卡在这里选择yes,那么可能会遇到循环登录的问题),如果不需要编译cuda代码时,在遇到安装cuda samples时也可以选择no。
3.2 安装完成后,执行一下命令,检查设备节点
此时并未遇到别的问题
3.3 在系统环境里设置cuda的路径
~$ sudo vim ~/.bashrc
插入文本(我的为64位系统,所以是lib64):
export PATH=/usr/local/cuda9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
使配置立即生效
~$ source .bashrc
验证是否安装成功,
~$ nvcc -V
我的结果显示
环境变量确定设置了,兵器重新启动几次,结果还是这样,所以按照要求执行了以下命令
~$ sudo apt install nvidia-cuda-toolkit
此时显示成功
4 安装cudNN
4.1到英伟达开发者官网https://developer.nvidia.com/rdp/cudnn-download下载对应cuda的Ubuntu16.04的cudNN文件,有几种安装方法,可参考https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html安装说明。