windows 配置pytorch环境

一、基础知识简介
（1）什么是CUDA
CUDA(ComputeUnified Device Architecture)，是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。
（2）什么是CUDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。
（3）CUDA与CUDNN的关系
CUDA是一个工作台，cuDNN是基于CUDA的深度学习GPU加速库，有了它才能在GPU上完成深度学习的计算，有了cuDNN才能在cuda上运行胜读神经网络。

二、安装cuda和cuDNN

【保姆级教程】Windows安装CUDA及cuDNN_windows安装cudnn-优快云博客

三、环境配置

3.1.安装好python和pycharm

直接百度搜索python和pycharm到官网直接安装就可以了

3.2.配置cuda环境

3.2.1 安装pytorch

到PyTorch官网，选择相应的安装方式

选择后，会出现pip或conda命令，如

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

 这个时候容易踩坑，要先根据自己电脑配置和自身需要选择CPU版本和GPU版本

3.2.2 我的安装方式

因为我在执行官网给的命令后，由于本地网络问题，无法正常安装，所以我采用先将对应版本的torch、torchvision、torchaudio下载到本地后进行安装，细节：

（1）先找到对应python版本，找到其中的Scripts文件夹，进入cmd命令：

        pip3.7 install "地址+xxx.whl"

（2）分别安装torch、torchvision、torchaudio

（3）进入pycharm查看是否安装成功

执行 print(torch.cuda.is_available()) ，显示True，即安装成功

四、安装过程遇到的其他问题

4.1 python多版本问题

（1）执行python --version，查看当前python版本

（2）切换python版本

这个把对应的python路径往顶部调就可以了，系统默认第一使用的python。 

4.2 cuda装错问题

主要是装成CPU了，网络问题包安装错了；在安装pytorch前，一定要将cuda和cudnn的版本对应好，不然后面会出问题。