(因为部分代码在低版本的torchvision上会报错,所以需要升级下torchvision)
一、先查询本机的python、torch和torchvision的版本号
查询python版本号【cmd下输入python就行,有确定的python环境的话就先用conda激活对应环境(否则就是默认的环境),再在其下输入python就行】:

我的环境的python版本3.8、torch版本1.6.0、torchvision版本0.1.6
找到官方提供的版本号匹配问题【https://pypi.org/project/torchvision/】,如下图所示,所以我将torchvision更新到0.7.0

二、torchvision具体的安装步骤:
1. 先在https://download.pytorch.org/whl/torch_stable.html找到适合自己版本whl文件,我的电脑没有cuda,所以直接下载的cpu版本,如下是我找到的需要的whl版本

2. 本地安装torchvision,使用pip方式,在环境下直接安装。
pip install whl文件名

3. 在python下使用下边代码核对torchvision版本信息。(显示0.7.0,表明安装成功。)
import torchvision
torchvision.__version__

最后,感谢这个帖子Windows10下GPU版本的pytorch1.6.0+torchvision0.7.0安装教程(支持CUDA10.2,含whl快速下载地址)
本文详细介绍了如何在Windows10上,针对Python 3.8、PyTorch 1.6.0的环境,通过本地whl文件将torchvision升级到0.7.0。步骤包括查询版本号、下载匹配的whl文件,以及使用pip进行安装和验证安装成功的操作。
643

被折叠的 条评论
为什么被折叠?



