windows下安装torchvision报错的解决方案

时间:2022-06-01 22:14:26

今晚再次多次尝试在win10下安装pytorch以及torchvision,终于把上次的坑给填了。来来回回装卸各种包有几十遍了,稍微记录一下,给后面遇到同样问题的人一点提醒。
首先,介绍下安装流程,比较简单,在Anaconda Prompt下运行如下官方指定命令。(这里用pip装,pytorch也可以用conda装,没什么区别,但torchvision只能用pip装,原因下面有说)

pip3 install pip3 install torchvision

上次的坑主要是:安装都成功了,并且import torch也没问题,但是在import torchvision的时候,出现了如下所示的错误信息:DLL load failed:找不到指定模块。

首先,我们得知道torchvision在conda里是装不了的,因为根本没有提供给win的资源,因此我们只能使用pip安装。

windows下安装torchvision报错的解决方案


安装完后,我们发现torchvision其实是有环境依赖的,其中有一条就是pillow的版本,而此次出错的问题主要就是这个PIL的问题。

我在网页上找资料的时候,没有找到直接的解决办法,但是有个博主在使用cv这个包的时候遇到了和我差不多的问题,最后他是通过把Pillow包的版本降到4.0.0后得以解决,因此我试着模仿,发现安装torchvision的时候自动给我更新到5.2.0,最后我试着把版本降回到4.1.1,也就是能够成功安装的最低版本,竟然得以解决。 pip uninstall Pillow pip install Pillow==4.1.1

因此最后得出的结论是:最新的Pillow包缺少了支持torchvision的东西,大家在遇到相同问题的时候,不妨把Pillow的版本降一降,希望能有所帮助。>.>

windows下安装torchvision报错的解决方案

标签:

原文地址:https://www.cnblogs.com/yhcheer/p/9532399.html