一次成功安装pytorch和torchvision

安装pytorch和torchvison

1、了解：

在安装pytorch前需要安装cuda和cudnn，该教程暂未涉及，后续会添加。
个人认为最好的安装方法是下载wheel轮子手动安装，直接用官方给的命令有时候网络啥的出现问题就安装失败了，而且安装慢。

2、第一步，下载对应的pytorch和torchvision的wheel包

下载地址：pytorch和torchvision的wheel下载地址。

2.1 下载torch的wheel文件

下载时要对应版本，一个是cuda版本一个是python版本。举个例子，我的cuda是11.7的，python是3.8的，如下所示：

注：前面的cu117代表cuda11.7版本的，cp38代表python3.8版本的，win_amd64是windows的，linux_x86是Linux系统的。下面还有几个torch也是cu117和cp38，只是torch本身版本不一样，个人理解没啥大问题，就看你自己想用哪个。

2.2 下载torchvision的wheel文件

下载torchvision和下载torch时差不多，主要在选择torchvision版本时要和torch一致，图片如下：

3、安装wheel文件

这两个包随便放哪都行，只要知道地址。
利用conda activate 进入自己的环境。
利用下面命令进行安装

pip install "torch的地址"
pip install "torchvision的地址"

这是我运行时的命令：

4、验证是否安装成功

利用conda activate进入虚拟环境
输入如下命令检测：

5、总结

以上就是本人安装pytorch中的过程，希望对各位有用。后续安装cuda和cudnn也会整理出来的。