此问题的核心是pytorch和torchvision的版本不对应
pytorch,torchvision,python 三者的对应关系来源于官方 github,链接如下:https://github.com/pytorch/vision#installation
我的解决方法如下:
- 打开Anaconda Prompt,进入python编辑器所在的虚拟环境
- 输入以下内容:
pip install torchvision==0.12.0 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
其中0.12.0替换为上图中对应的内容即可
3. 重启VSCODE