参考网址:https://blog.csdn.net/zhuangwu116/article/details/81063234
(1)下载安装文件:
下载cuda9.0 runfile 文件 下载地址
下载cudnn deb文件一共有两个文件,一个库文件一个开发文件,下载cudnn需要注册账号 下载地址
(2)安装cuda9.0
在Ubuntu桌面系统下按快捷键:
Ctrl+Alt+F2
进入终端模式,在终端模式关闭桌面服务,命令:
sudo service lightdm stop
不关闭桌面服务无法安装cuda9.0,并创建文件,在/etc/modprobe.d/blacklist-nouveau.conf
内容如下:
blacklist nouveau
options nouveau modeset=0
之后运行命令让它生效,这是为了禁用第三方cuda驱动。
sudo update-initramfs -u
安装依赖包:
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
运行runfile脚本文件安装:
sudo sh cuda_9.0.176_384.81_linux.run
这里需要注意的是安装过程会咨询你是否安装OpenGL,要输入N,如果选择Y,到时候Ubuntu登录系统界面,输入密码会出现循环登录问题。其余选项都是Y
安装完成之后运行:
sudo service lightdm start
启动桌面服务程序。
注意:
1.这里顺便说一下如果已经出现循环登录问题的按照上面所有的步骤重新运行runfile文件安装即可它会直接覆盖以前安装的所有库,已经配置环境变量cuda9.0的,安装cudnn的,安装tensorRT的,已经配置tensorRT的无需在安装配置,也不需要重新安装ubuntu系统。
2.在已经安装cuda9.0,cudnn,tensortde,在关机重新开机的,或者重启时,可能会出现登录界面显示的过大,不正常,直接运行runfile文件重新安装设备驱动即可,其他不用重新安装配置,也就是说询问你是否安装设备驱动输入y,其他都输入n即可,也就是说我们只要安装驱动,不需要重新安装cuda9.0库,cudnn,tensorT,因为之前已经安装过了。
配置环境
打开~/.bashrc文件,在该文件的结尾添加:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
或者
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
配置玩环境变量,使其生效:
source ~/.bashrc
(3)安装cudnn比较简单无需配置环境变量,只需要2条安装命令即可,这两条命令顺序不可颠倒,否则安装不成功:
sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.0_amd64.deb