目录
一 Ubuntu Nvidia显卡驱动安装
1.1 先得禁用ubuntu自带的显卡驱动 nouveau
1.2 安装驱动
二 cuda9.0 安装(deb安装)
2.1 下载地址:(选deb)
2.2 下载完毕后进入下载的文件夹下依次输入:
2.3 设置环境变量
三 cudnn 安装
3.1 下载
四 测试
附:(参考链接)
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html(cuda对应的nvidia版本)
依次输入如下的命令:
ll /etc/modprobe.d/blacklist.conf
sudo chmod 666 /etc/modprobe.d/blacklist.conf
sudo gedit /etc/modprobe.d/blacklist.conf
在打开的文件末尾加入如下几行:
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb
然后输入:
sudo chmod 644 /etc/modprobe.d/blacklist.conf
sudo update-initramfs -u
*********重启电脑*********
查看是否屏蔽(若无打印的内容,
则说明屏蔽成功
)
lsmod | grep nouveau
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-418
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
**********重启电脑************
sudo apt-get update
sudo apt-get upgrade
**********检查是否成功*************
watch nvidia-smi
二
cuda
9.0
安装(
deb
安装
)
2.1
下载地址:(选
deb
)
Please download from here
2.2
下载完毕后进入下载的文件夹下依次输入:
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
*****
重启电脑
*****
ls /dev/nvidia*
结果若显示:
则表示成功 !
sudo gedit /etc/profile
打开文件,在末尾处加入命令如下(针对64位系统):
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
*****
重启电脑
*****
验证驱动版本:
cat /proc/driver/nvidia/version
结果如下所示,则表示成功安装
验证cuda Toolkit:
nvcc -V
成功的话会输出版本信息!(至此,cuda安装完毕,下马安装cudnn,一定要注意对照着cuda版本进行下载安装)
ps:如果出现上述的结果 说明cuda安装的时没有问题的,如果出现后续的测试出现(段错误:核心已转储)错误,那么应该是你的cuda与显卡驱动版本没有对应好,显卡驱动可以比cuda高 但cuda版本不能比显卡驱动高....
download from here
(ps: 在此记录一下自己的密码防止忘记(读者请跳过此ps内容) sz*********hb520!)
下载完毕后 ,cd 到 放置文件的目录下执行如下命令(或在该目录下打开终端):
tar -xzvf cudnn-9.0-linux-x64-v7.4.1.5.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
至此,ubuntu下的nvidia驱动+cuda+cudnn配置完毕!
可安装 pytorch 进行测试 download from here (注意版本的对应)
安装完成后 ,测试一下:
>>import torch
>>torch.cuda.is_available()
true
PS:当然也有别的测试方法...
1 卸载现有的显卡驱动
sudo apt-get remove --purge nvidia-*
2 卸载和重新安装显卡驱动不影响 已经安装的cuda 和cudnn....
3 cuda的环境配置,网上的有不同版本,但效果应该都可以....
4 安装显卡驱动的时候,最好安装高版本的,这样不会受cuda版本的影响...
附:(参考链接)
https://www.jianshu.com/p/71bc5f02ecd2
https://www.cnblogs.com/gongdiwudu/p/7653663.html
https://blog.csdn.net/qq_33200967/article/details/80689543#CUDA_71 (有关卸载cuda&cudnn)