在下载项目需要的torchvision后,在Pycharm上运行下面这段代码,总是会报上面这个错误,根据报错的提示,我猜测应该是torchvision出了问题:
import torch
from torch import nn
from torch.nn import init
import numpy as np
import sys
sys.path.append("..")
import d2lzh_pytorch as d2l
print(torch.__version__)
一开始我以为是包安装的路径问题,所以将项目的虚拟环境从电脑默认的环境改成了新建的项目环境,但是还是会报错,于是我在网上查了一些其他博主的解决方案,考虑应该是安装的torch版本和torchvision版本不对应的问题,torch和torchvision对应版本查询官网后了解到,torch1.8.0版本对应的torchvision版本应该是0.9.0。于是卸载了原本的torch,重新安装了一个新的torch版本与torchvision版本对应,可以在代码块中输入查询版本,最终就不再报错了!
卸载torch代码:
pip uninstall torch
安装torch代码:
pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
成功安装torch返回的结果:
查询torch及torchvision版本代码:
import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
|