cuda9支持的linux内核,Ubuntu16.04安装CUDA9.0+CUDNN

写在前面:如果是新装的Ubuntu系统,请先换源。可以考虑清华源:

将/etc/apt/sources.list中内容改为

# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse

# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse

# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse

# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse

# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse

# 预发布软件源,不建议启用

# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

安装CUDA

安装CUDA是坑最多的,也是最麻烦的。

1.需要更换Linux内核,我的当前内核版本为4.15,需要换到4.4

a.查看当前安装的内核版本

dpkg -l|grep linux-image

b.查看可以更新的内核版本

sudo apt-cache search linux-image

c.安装新内核

sudo apt-get install linux-image-4.4.0-77-generic linux-image-extra-4.4.0-77-generic

d.更新grub引导

sudo update-grub

e.重启按Esc或Shift+Tab选择Ubuntu高级选项,进入4.4的内核

f.卸载不要的内核

sudo apt-get purge linux-image-xx-xx-generic linux-image-extra-xx-xx-generic

g.更新grub引导

sudo update-grub

2.安装缺少的依赖库

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

3.禁用nouveau驱动

a.在/etc/modprobe.d中创建文件blacklist-nouveau.conf

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

很多时候vi不好用,考虑安装vim

b.在文件中输入:

blacklist nouveau

options nouveau modeset=0

c.执行生效

sudo update-initramfs –u

d.查看是否已关闭

lsmod | grep nouveau

若无内容输出,则成功禁止

4.进入命令行界面

重新启动,到登录界面时按Ctrl+Alt+F1

5.关闭图形化界面

sudo service lightdm stop

6.运行下载好的CUDA程序

选择CUDA9.0,此版本兼容性最好,选择runfile安装

sudo sh cuda.run

按Ctril+c跳过协议阅读部分

在安装选项中,不要安装OpenGL,不然后循环停留在登录界面

因为我电脑本没有安装nvidia显卡驱动,装CUDA时顺便装上了

最后显示installed,表示成功了

7.重新启动图形化界面

sudo service lightdm start

按Ctrl+Alt+F7回到登录界面

8.设置环境变量

在~/.bashrc中加入

export PATH=/usr/local/cuda-9.0/bin:$PATH

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

7.验证CUDA安装完成

a.nvcc -V

b.nvidia-smi

安装CuDnn

1.下载对应版本的cudnn

2.解压

tar -zxf cudnn-xxxx.tgz

3.把include/ 和 lib64/目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64里面

sudo cp -P include/cudnn.h /usr/local/cuda-9.0/include

sudo cp -P lib64/libcudnn* /usr/local/cuda-9.0/lib64

sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*

至此,CUDA与CuDnn安装完成。

你可能感兴趣的:(cuda9支持的linux内核)