关于 Detectron2 的导入错误

Question

我尝试使用

Detectron2

进行语义分割，但是当我运行程序时出现了一些棘手的错误。看来我的环境可能存在一些问题。

有谁知道如何解决吗？

导入错误：无法从“torch.fx._symbolic_trace”导入名称“is_fx_tracing”（/home/eric/anaconda3/envs/detectron_env/lib/python3.9/site-packages/torch/fx/_symbolic_trace.py）

Answer 1

这似乎是 detectorron2 最新提交的问题，您可以在安装时使用 detectorron2 之前的提交来避免此错误。

pip install 'git+https://github.com/facebookresearch/detectron2.git@5aeb252b194b93dc2879b4ac34bc51a31b5aee13'

该问题已在 detectorron2 的最新提交中得到解决

Answer 2

@Mohan Ponduri 的解决方案对我有用。谢谢。似乎是新安装 Detectron2 的问题。我还可以在 conda 环境中使用 detectorron2，以防万一有人想知道

Answer 3

我使用了@Mohan Ponduri 建议的方法。但我收到以下错误。请告诉我我需要做什么。

引发运行时错误（CUDA_MISMATCH_MESSAGE.format（cuda_str_version，torch.version.cuda）） 运行时错误： 检测到的 CUDA 版本 (11.5) 与用于编译的版本不匹配 PyTorch (12.1)。请确保使用相同的 CUDA 版本。

  [end of output]