当前位置:   article > 正文

pytorch安装+出现ModuleNotFoundError: No module named ‘torchvision‘的解决办法_modulenotfounderror: no module named 'torchvision

modulenotfounderror: no module named 'torchvision

一、PyTorch安装

anaconda中安装pytorch,如果仅使用cpu。登录pytorch官网Py火炬 (pytorch.org)

如果需要更低版本,点击“Previous versions of PyTorch”,选择相应版本

上图我选择的是pytorch=1.1.0 版本。如果选择cuda版本还需要另外安装cuda,并且电脑需要有GPU或者租服务器。

二、ModuleNotFoundError: No module named ‘torchvision‘

1.首先,使用conda list,查看torchvision是否安装成功。我的就是只安装了pytorch

解决办法:只需要重新安装torchvision,安装版本选择参照本文的一、部分

如果安装失败,尝试先将pytorch卸载。然后重新安装

2.如果conda list后,有pytorchvision,那就看一下是不是pytorch和pytorchvision版本不匹配。需要重新下载。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/849485
推荐阅读
相关标签
  

闽ICP备14008679号