Pytorch安装框架流程/总结

本文使用的在线安装，对安装Pytorch的步骤作总结，整理框架流程，防止踩坑。安装具体细节可以参考别的博主

        安装中涉及到Anaconda，Python，Cuda，CuDnn，Pytorch。

        其中Cuda，CuDnn，Pytorch不兼容严重，一定要找准版本，我的建议是先确定Pytorch版本再确定Cuda，CuDnn版本

离线库网址：

使用离线安装比较麻烦，对python的版本也是有要求的，列如：

指的是这个torch版本为1.10.0，需要Cuda11.3，Python3.6版本，windows64位。

此外还需要添加对应的 torchvision和torchaudio的库

如有不对，请大佬指教

一.安装Anaconda

1.1准备工作

安装Python环境，推荐环境3.8

安装Pycharm，版本是2020.1

安装Anaconda，推荐版本Anaconda为2022.10

解压完后，分别添加Python和Anaconda的相关路径

安装后使用win+r，创建自己的虚拟环境，并指定Python版本，如果没有Python环境。可以自己加一下很简单

1.2常用按键

安装好Anaconda后，需要创建自己的虚拟环境

创建虚拟环境

conda create -n 环境名 python=3.9

进入虚拟环境

conda activate 环境名

退出虚拟环境

conda deactivate

查看conda下的虚拟环境

conda info --e

列出当前环境下的库

conda list

查看Cuda版本

nvcc -V

查看cudnn版本

conda activate yolov5

python

import torchvision

print(torchvision.__version__)

二.安装Cuda和Cudnn

1.1 Cuda的版本

在安装Cudah前，一定要先确认自己要多少版本的pytorch库

        比如安装1.6版本的pytorch库，就需要去安装10.2版本的Cuda

        再去找对应的Cudnn版本，例如：10.2版本的Cuda对应的是7.6.5的Cudnn

如下是Cuda版本对应Pytorch版本

1.2 torchvison 版本