运维开发网
广告位招商联系QQ:123077622
 
广告位招商联系QQ:123077622

64位Ubuntu14.04+GTX960安装CUDA8.0及cudnn

运维开发网 https://www.qedev.com 2020-03-20 08:23 出处:网络 作者:运维开发网整理
0.检查gpu,系统版本和内核是否支持,一般没问题。 部分计算机需要在BIOS中关闭secure boot

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 --purge nvidia-* 卸载驱动
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl 卸载cuda

扫码领视频副本.gif

0

精彩评论

暂无评论...
验证码 换一张
取 消

关注公众号