安装CUDA

先安装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

非原创,记录下来只是为了方便日后查找问题,参考链接:

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值