安装pytorch库的步骤

目前开始接触python学习，主要涉及神经网络相关计算，以下是安装pytorch库过程中的遇到的问题与解决方案

在添加torch库时出现了pytorch版本与cuda、cudnn版本不对应问题，导致添加了后依旧出现了如下语句：

ModuleNotFoundError: No module named torch

这里先给出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快很多。

为了一劳永逸，最终选择在Anaconda下安装pytorch环境。

Anaconda下载完成后，首先在cmd命令中输入

nvidia-smi

即可查询当前显卡对应的CUDA版本

如本机显卡对应的CUDA版本为12.3。在此基础上在NVIDIA官网进行下载，具体网址如下

https://developer.nvidia.com/cuda-12-3-0-download-archive

CUDA工具包的安装和下载可以参照

CUDA安装和检测【全】（nvcc命令找不到的解决办法）_深度学习_可靠的豆包蟹同志-GitCode 开源社区

在此基础上下载CUDNN，并安装至已加载好的CUDA中，参照操作如下

安装CUDA_nf5468a5 install cuda-CSDN博客

最后在anaconda中安装pytorch，参照操作如下

在Anaconda下安装Pytorch的超详细步骤「建议收藏」-腾讯云开发者社区-腾讯云

在此之后，在pycharm数据库的设置选项里，选择项目-python解释器-添加解释器-添加本地解释器-conda环境，即可配置成功。