cuda安装

--cuda and cudnn

Posted by Jieson on October 31, 2019

本次操作都是基于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 image 手动搜索所有版本: https://www.geforce.cn/drivers image 选择了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 image image 选择自己需要安装的版本,下载.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 image image image

选择自己需要的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 保存文件,执行该命令,使配置生效