《动手学深度学习(PYTORCH版)》第3章导入torchvision后,报错无法定位程序输入点

《动手学深度学习(PYTORCH版)》第3章导入torchvision后,报错无法定位程序输入点_第1张图片
在下载项目需要的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返回的结果:
《动手学深度学习(PYTORCH版)》第3章导入torchvision后,报错无法定位程序输入点_第2张图片

查询torch及torchvision版本代码:

import torch
print(torch.__version__)

import torchvision
print(torchvision.__version__)

安装成功之后查询版本的结果,就不再报错了。
《动手学深度学习(PYTORCH版)》第3章导入torchvision后,报错无法定位程序输入点_第3张图片

你可能感兴趣的:(《动手学深度学习(PYTORCH版)》第3章导入torchvision后,报错无法定位程序输入点)