我一直在关注这个PyTorch tutorial,用我自己的数据集微调Mask R-CNN模型。理论上看起来很简单,但在实践中,我遇到了一些关于torch
和torchvision
的问题。
首先,当我运行代码时,似乎有些模块没有被识别。我尝试了不同的最新版本的torch
,甚至直接从GitHub克隆了torch
存储库。尽管如此,我还是得到了一个错误,比如:
---> 16 if torch._running_with_deploy():
AttributeError: module 'torch' has no attribute '_running_with_deploy'
其次,我还从GitHub克隆了torchvision
存储库,希望它能解决我的问题。但是,我仍然遇到错误,例如:
ImportError: cannot import name '_meta_registrations' from 'torchvision' (C:\Users\anaconda33\lib\site-packages\torchvision\__init__.py)
我花了很多时间在网上寻找解决方案,但似乎没有什么能真正解决我的问题。我有点不知所措,真的很感激你的帮助。
如果有人以前遇到过这些问题,或者对如何解决这些问题有建议,我洗耳恭听。
提前感谢!
到目前为止,我一直在努力解决这些问题:
- 已在
torch
和torchvision
的多个版本之间切换,以查看是否存在兼容性问题。 - 直接从GitHub克隆了
torch
和torchvision
存储库,认为最新的代码可能会解决这个问题。 - 检查了我的环境变量和路径,以确保Python在正确的地方查找。
1条答案
按热度按时间goqiplq21#
我通过安装
pytorch
的nightly build运行了本教程(本教程开头的注解提到**>=0.16或nightly**是必需的)。下面是我使用的命令:第二个
pip
命令安装pytorch
的夜间版本,它是从https://pytorch.org/生成的。您可以在这里指定要生成安装命令的操作系统、pytorch
版本、包管理器。