ubuntu16.04+nvidia显卡驱动+cuda9.0(deb安装法)+cudnn7.4

目录

一 Ubuntu Nvidia显卡驱动安装

1.1  先得禁用ubuntu自带的显卡驱动 nouveau

1.2 安装驱动

二 cuda9.0 安装(deb安装)

2.1 下载地址:(选deb)

2.2 下载完毕后进入下载的文件夹下依次输入:

2.3 设置环境变量

三  cudnn 安装

3.1 下载

四 测试

附:(参考链接)


Ubuntu Nvidia显卡驱动安装

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html(cuda对应的nvidia版本)

1.1  先得禁用ubuntu自带的显卡驱动 nouveau

依次输入如下的命令:

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

1.2 安装驱动

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

cuda9.0 安装(deb安装

2.1 下载地址:(选deb

 Please  download from here

ubuntu16.04+nvidia显卡驱动+cuda9.0(deb安装法)+cudnn7.4_第1张图片

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*

结果若显示:

则表示成功 !

2.3 设置环境变量

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版本不能比显卡驱动高....

三  cudnn 安装

3.1 下载

      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 (注意版本的对应)

ubuntu16.04+nvidia显卡驱动+cuda9.0(deb安装法)+cudnn7.4_第2张图片

安装完成后 ,测试一下:

>>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)

你可能感兴趣的:(gpu,cuda,pytorch,ubuntu,深度学习环境搭建)