官网文档地址:
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
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.f2er.com/ubuntu/350040.html