先安装cuda,再安装cudnn
一、CUDA、CUDNN介绍
1、什么是CUDA
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
2、什么是CUDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
3、CUDA与CUDNN的关系
CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。
想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。
二、CUDA安装
1、电脑显卡支持的最高CUDA驱动版本怎么查看?
方法:win键——所有应用——NVIDIA control panel。
在NVIDIA控制面板中依次点击,帮助——系统信息——组件。
如图,楼主电脑显卡为最高支持11.1.114版本的CUDA驱动。
前往 Nvidia 的 CUDA 官网:CUDA Toolkit Archive | NVIDIA Developer ,笔者下载 CUDA 11.1.1 版本:
wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run
三、安装CUDnn
1、注册NVDIA账号
官网下载cuda对应版本的cudnn:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111
这里我们选择的是8.0.5.39版本的cudnn
下载一下三个文件:
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.1_amd64.deb
查看cudnn版本:
dpkg -l | grep cudnn
如下图:
参考连接
https://blog.csdn.net/Sihang_Xie/article/details/127347139
四、安装显卡驱动
(1)更新软件源
在其他站点中选择清华源,更新后终端执行
sudo apt-get update
sudo apt-get upgrade
(2)安装驱动:
菜单栏选择附加驱动,会进行自动搜索,选择一个版本的专有驱动,点击应用更改:
方法1:
然后执行命令,来完成安装:自动匹配版本:
sudo ubuntu-drivers autoinstall
4.1、nvidia-smi报错:NVIDIA-SMI has failed because it couldn‘t communicate with the NVIDIA driver 原因及避坑解决方案)
第一步,安装dkms:
sudo apt-get install dkms
第二步,查看本机连接不上的驱动版本:
ls -l /usr/src/
如果安装失败了,请跟者本文走>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
此时,如果你的gcc(尽量大于7.3版本)版本过低,那么上述命令
sudo dkms install -m nvidia -v 470.141.03
失败的原因就找到了,查看现有的gcc版本:
```c
gcc --version
470.141.03
这里我讲解安装gcc7.5版本的过程:
第一步,下载gcc,输入命令:
cd /usr/local/src/
wget http://ftp.gnu.org/gnu/gcc/gcc-7.5.0/gcc-7.5.0.tar.xz tar xvf
遇到解析失败问题,
tar xvJf gcc-7.5.0.tar.xz
第二步,安装gcc的依赖软件,gcc的软件包内提供了自动下载需要软件的脚本,命令如下:
cd gcc-7.5.0/
./contrib/download_prerequisites
第三步,生成Makefile文件,输入命令:
./configure -enable-checking=release -enable-languages=c,c++ -disable-multilib
第四步,编译gcc,命令如下:
make -j4
ls /usr/local/bin | grep gcc
多核电脑可以添加 “-j4” ,make对多核处理器的优化选项,此步骤非常耗时 !
第五步,安装gcc,命令如下:
make install
sudo dkms install -m nvidia -v 470.141.03
五、卸载显卡驱动
1.执行卸载
sudo apt-get --purge remove nvidia*
sudo apt autoremove
原文链接:https://blog.csdn.net/nizhenshishuai/article/details/123873453
非原创,记录下来只是为了方便日后查找问题,参考链接: