cuda、cudnn、pytorch相关配置问题

环境配置相关

​

1、什么是CUDA

CUDA(ComputeUnified Device Architecture)，是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。

2、什么是CUDNN

NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型，而不是简单调整性能，同时还可以在GPU上实现高性能现代并行计算。

3、CUDA与CUDNN的关系

CUDA看作是一个工作台，上面配有很多工具，CUDNN是基于CUDA的深度学习GPU加速库，有了它才能在GPU上完成深度学习的计算。它就相当于工作台上的工具，但是CUDA这个工作台买来的时候，并没有送工具。想要在CUDA上运行深度神经网络，就要安装CUDNN。就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作，工作速度相较CPU快很多。

4.安装

4.1安装CUDA和CUDNN

上图横线处就是支持的CUDA最高版本

根据CUDA的版本，安装CUDNN

4.2安装pytorch

注意！！！

在安装之前先用conda创建虚拟环境，在虚拟环境里面安装

根据自己的CUDA版本在Pytorch官网https://pytorch.org/看

在安装前，先使用清华的镜像源（国内下载速度更快）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

（恢复默认源）

conda config --remove-key channels

(查看当前源)

conda config --show channels

替换源之后我们要删掉-c pytorch的命令（使用默认源来安装的意思），故实际安装的命令为：

conda install pytorch torchvision torchaudio pytorch-cuda=11.7  -c nvidia

之后进入python环境，输入下列验证是否安装成功