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

发布时间:2024年01月17日

一、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版本不匹配。需要重新下载。

文章来源:https://blog.csdn.net/weixin_44966620/article/details/135635268
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。