一次成功安装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的,如下所示:
一次成功安装pytorch和torchvision_第1张图片
注:前面的cu117代表cuda11.7版本的,cp38代表python3.8版本的,win_amd64是windows的,linux_x86是Linux系统的。下面还有几个torch也是cu117和cp38,只是torch本身版本不一样,个人理解没啥大问题,就看你自己想用哪个。

2.2 下载torchvision的wheel文件

下载torchvision和下载torch时差不多,主要在选择torchvision版本时要和torch一致,图片如下:
一次成功安装pytorch和torchvision_第2张图片

3、安装wheel文件

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

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

这是我运行时的命令:
在这里插入图片描述
一次成功安装pytorch和torchvision_第3张图片

4、验证是否安装成功

利用conda activate进入虚拟环境
输入如下命令检测:
在这里插入图片描述
在这里插入图片描述

5、总结

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

你可能感兴趣的:(深度学习,pytorch,深度学习,python)