写在前面:如果是新装的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安装完成。