使用 PyTorch 教程微调 Mask R-CNN 的问题：Torch 和 Torchvision 的模块错误

Question

我一直在按照这个 PyTorch 教程使用我自己的数据集微调 Mask R-CNN 模型：PyTorch 教程 (https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html)。在纸面上，这似乎很简单，但在实践中，我遇到了 torch 和 torchvision 的几个问题。

首先，当我运行代码时，似乎有些模块无法被识别。我尝试过不同的最新版本的 torch，甚至直接从 GitHub 克隆了 torch 存储库。尽管如此，我还是收到如下错误： ---> 16 如果 torch._running_with_deploy(): AttributeError：模块“torch”没有属性“_running_with_deploy”

其次，我还从 GitHub 克隆了 torchvision 存储库，希望它能解决我的问题。但是，我仍然遇到错误，例如： 导入错误：无法从“torchvision”导入名称“meta_registrations”（C：\ Users naconda33 \ lib \ site-packages orchvision_init.py）

我花了相当长的时间在网上寻找解决方案，但似乎没有什么能真正解决我的问题。我有点不知所措，非常感谢一些帮助。

如果有人以前遇到过这些问题或对如何解决这些问题有建议，我洗耳恭听。

提前致谢！

到目前为止我为解决这些问题所做的尝试：

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

Answer 1

我通过安装

pytorch

的夜间版本来管理运行本教程（教程开头的注释提到需要 >=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

构建、包管理器生成安装命令。