微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Ubuntu16.04 下安装Cuda9.1+Cudnn7

官网文档地址:
http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#ubuntu-installation

一、系统需求

1. 查显卡型号是否支持

lspci | grep -i nvidia

找到VGA compatible controller型号,并到Cuda网站查询
https://developer.nvidia.com/cuda-gpus

2. Linux 版本验证

3. gcc 版本验证

4. Kernel Headers和开发包验证

目前Ubuntu16.04这些是支持的。

5. 选择安装方式

https://developer.nvidia.com/cuda-downloads
我使用的是离线下载。

6. 如果之前已经安装了Cuda,则卸载掉。

7. 禁用nouveau

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

内容

blacklist nouveau options nouveau modeset=0

更新系统:

sudo update-initramfs -u

重新启动,验证下面一句无输出则为成功:

lsmod | grep nouveau

禁用图形界面

sudo service lightdm stop

按ctrl+alt+f1进入命令行界面。
上面操作如果不行,则直接移除:

cd /lib/modules/4.10.0-28-generic/kernel/drivers/gpu/drm/nouveau 
sudo rm -rf nouveau.ko 
sudo rm -rf nouveau.ko.org

8. 安装nvidia驱动

下载驱动:
http://www.geforce.cn/drivers

卸载原驱动

sudo apt-get remove --purge nvidia-*

我在操作的时候,不幸遇到:

ubuntu 16 ImportError: No module named 'ConfigParser'

处理方法:把python先切回python2

rm /usr/bin/python
ln -s /usr/bin/python2 /usr/bin/python
# 删除python-configparser,再通过aptitude来安装
sudo apt-get purge --auto-remove python-configparser
sudo aptitude install python-configparser

再卸载原驱动。

安装新下载的驱动

sudo chmod a+x NVIDIA-Linux-x86_64-384.111.run //获取权限
sudo ./NVIDIA-Linux-x86_64-384.111.run -no-x-check -no-nouveau-check -no-opengl-files //安装驱动
sudo reboot
service lightdm restart
  • –no-x-check 安装驱动时关闭X服务
  • –no-nouveau-check 安装驱动时禁用nouveau
  • –no-opengl-files 只安装驱动文件,不安装OpenGL文件

    验证:

nvidia-smi

安装cuda

sudo sh ./cuda_9.1.85_3

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 387.26?
要选否。
安装认位置: /usr/local/cuda-9.1
samples位置: /home/admin1

下载安装Cudnn7.0.5 For Cuda9.1

https://developer.nvidia.com/rdp/cudnn-download#a-collapse704-9

测试cuda demo

cd /home/admin1
mkdir test
cd test
cp -r /home/admin1/NVIDIA_CUDA-9.1_Samples/ .
cd NVIDIA_CUDA-9.1_Samples
make

原文地址:https://www.jb51.cc/ubuntu/350040.html

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐