0.检查gpu,系统版本和内核是否支持,一般没问题。
- 部分计算机需要在BIOS中关闭secure boot
- 如果出现黑屏问题,请看这里
1.gcc和g++调整:安装cuda对应的gcc和g++并进行软链接,本教程cuda对应4.8
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
sudo apt-get install -y gcc-4.8
sudo apt-get install -y g++-4.8
sudo su
cd ../../usr/bin
ln -s /usr/bin/g++-4.8 /usr/bin/g++ -f
ln -s /usr/bin/gcc-4.8 /usr/bin/gcc -f
2.安装32位兼容包。因为安装二进制安装包时,提示是否安装32bit兼容包,所以,最好安装上32兼容包,以后操作系统上安装32bit软件也方便。
sudo apt-get install -y lib32z1
3.禁用nouveau
lsmod | grep nouveau 查看nouveau是否使用,无输出则意味着被禁用或没有
有则执行sudo vim /etc/modprobe.d/blacklist-nouveau.conf, 输入
blacklist nouveau
options nouveau modeset=0
保存退出
在终端执行sudo update-initramfs -u
重启电脑
4.安装CUDA
sudo service lightdm stop 关闭图形界面,按ctrl+alt+F1进入tty1并登录
进入run文件所在目录
chmod 777 cuda_8.0.61_375.26_linux.run
sudo ./cuda_8.0.61_375.26_linux.run 需要时间读取许可,按q退出阅读许可
依次接受许可,安装driver:yes,安装toolkit和samples:yes,中间选择及安装路径直接Enter选择默认
sudo service lightdm start 打开图形界面
执行vim ~/.bashrc,在文件最后加入
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
执行source ~/.bashrc使配置的环境变量生效
nvidia-smi 出现GPU相关信息表示驱动安装成功
执行nvcc --version查看cuda版本信息
5.测试cuda的Samples
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
保证编译没有error,执行deviceQuery输出信息的最后类似以下:
deviceQuery,CUDA Driver = CUDART,CUDA Driver Version = 8.0,CUDA Runtime Version = 8.0,NumDevs = 1,Device0 = GeForce GTX 960
Result = PASS
6.cudnn安装
tar xvzf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/*.* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
7.卸载
sudo apt-get remove
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl 卸载cuda
原文链接:https://www.f2er.com/ubuntu/352722.html