python 问题使用PyTorch微调掩码R-CNN教程:Torch和Torchvision的模块错误

6rqinv9w  于 2023-09-29  发布在  Python
关注(0)|答案(1)|浏览(131)

我一直在关注这个PyTorch tutorial,用我自己的数据集微调Mask R-CNN模型。理论上看起来很简单,但在实践中,我遇到了一些关于torchtorchvision的问题。
首先,当我运行代码时,似乎有些模块没有被识别。我尝试了不同的最新版本的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)

我花了很多时间在网上寻找解决方案,但似乎没有什么能真正解决我的问题。我有点不知所措,真的很感激你的帮助。
如果有人以前遇到过这些问题,或者对如何解决这些问题有建议,我洗耳恭听。
提前感谢!
到目前为止,我一直在努力解决这些问题:

  • 已在torchtorchvision的多个版本之间切换,以查看是否存在兼容性问题。
  • 直接从GitHub克隆了torchtorchvision存储库,认为最新的代码可能会解决这个问题。
  • 检查了我的环境变量和路径,以确保Python在正确的地方查找。
goqiplq2

goqiplq21#

我通过安装pytorch的nightly build运行了本教程(本教程开头的注解提到**>=0.16或nightly**是必需的)。下面是我使用的命令:

!pip uninstall -y torch
!pip install --pre torch torchvision  --index-url https://download.pytorch.org/whl/nightly/cpu

第二个pip命令安装pytorch的夜间版本,它是从https://pytorch.org/生成的。您可以在这里指定要生成安装命令的操作系统、pytorch版本、包管理器。

相关问题