本次操作都是基于ubuntu16.04的操作,安装cuda9.0,cudnn7.3.1
环境
系统:ubuntu16.04
显卡型号:GeForce GTX 1080
安装步骤
1、如果以安装显卡驱动,先卸载原来的驱动,如果没有则跳过
sudo apt-get remove --purge nvidia*
如果是通过runfile文件安装,可通过原始安装文件卸载
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-430.50.run --uninstall
2、下载驱动
官网地址: https://www.nvidia.com/Download/index.aspx 手动搜索所有版本: https://www.geforce.cn/drivers 选择了430.50版本,安装cuda10也不需要更换驱动 给文件添加执行权限
sudo chmod +x NVIDIA-Linux-x86_64-430.50.run
3、禁用nouveau驱动(禁用nouveau第三方驱动,之后也不需要改回来) 打开blacklist.conf文件
sudo gedit /etc/modprobe.d/blacklist.conf
添加
blacklist nouveau
options nouveau modeset=0
保存文件后执行
sudo update-initramfs -u
然后重启机器,执行:lsmod | grep nouveau。如果没有屏幕输出,说明禁用nouveau成功。
4、关闭X-Window服务
sudo service lightdm stop # 这会关闭图形界面,安装完后开启图形界面
sudo service lightdm start # 开启图形界面
关闭图形界面后,按Ctrl-Alt+F1进入命令行界面,输入用户名和密码登录即可,也可以在关闭图形界面前开启 ssh远程连接服务,通过另一台机器远程操作 5、安装驱动
sudo ./NVIDIA-Linux-x86_64-430.50.run
一般正规机器就能够顺利安装成功。
参数说明:
- –no-opengl-files:表示只安装驱动文件,不安装OpenGL文件。这个参数不可省略,否则会导致登陆界面死循环,英语一般称为”login loop”或者”stuck in login”。
- –no-x-check:表示安装驱动时不检查X服务,非必需。
- –no-nouveau-check:表示安装驱动时不检查nouveau,非必需。
- -Z, –disable-nouveau:禁用nouveau。此参数非必需,因为之前已经手动禁用了nouveau。
- -A:查看更多高级选项。
必选参数解释:因为NVIDIA的驱动默认会安装OpenGL,而Ubuntu的内核本身也有OpenGL、且与GUI显示息息相关,一旦NVIDIA的驱动覆写了OpenGL,在GUI需要动态链接OpenGL库的时候就引起问题。
安装成功后,
nvidia-smi #若列出GPU的信息列表,表示驱动安装成功
6、cuda 安装
官网: https://developer.nvidia.com/cuda-toolkit-archive 选择自己需要安装的版本,下载.run文件,并把相关补丁下载下来,如cuda9.0有四个补丁文件
cuda_9.0.176_384.81_linux.run
cuda_9.0.176.1_linux.run
cuda_9.0.176.2_linux.run
cuda_9.0.176.3_linux.run
cuda_9.0.176.4_linux.run
安装cuda
sudo ./cuda_9.0.176_384.81_linux.run
在安装过程中,当询问是否安装驱动时请选择否
也可以加上参数,
- –no-opengl-libs:表示只安装驱动文件,不安装OpenGL文件。必需参数,原因同上。注意:不是-no-opengl-files。
- –uninstall (deprecated):用于卸载CUDA Driver(已废弃)。
- –toolkit:表示只安装CUDA Toolkit,不安装Driver和Samples。
- –help:查看更多高级选项
然后把剩下的补丁文件安装
sudo ./*.run
7、安装cudnn
官网: https://developer.nvidia.com/cudnn
选择自己需要的cudnn版本下载,下载完成后解压文件
tar -zxvf cudnn-9.0-linux-x64-v7.3.1.20.tgz # 生成一个cuda文件夹
# 将文件拷贝到cuda文件夹,并修改文件权限
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
8、配置环境变量
vim ~/.bashrc # 打开文件,将cuda环境添加进去
###
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-9.0/bin:$PATH
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
###
source ~/.bashrc 保存文件,执行该命令,使配置生效