如何修复“导入错误：无法在 Python 中导入名称‘numpy_type_map’”？

Question

我已按照Detectron 中的说明进行操作，并对其进行了多次配置：代码按预期进行编译。 在运行代码时，我收到此错误：

    Traceback (most recent call last):
    File "tools/train_net_step.py", line 21, in <module>
    import nn as mynn
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/nn/__init__.py", line 2, in <module>
    from .parallel import DataParallel
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/nn/parallel/__init__.py", line 3, in <module>
    from .data_parallel import DataParallel, data_parallel
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/nn/parallel/data_parallel.py", line 4, in <module>
    from .scatter_gather import scatter_kwargs, gather
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/nn/parallel/scatter_gather.py", line 8, in <module>
    from torch.utils.data.dataloader import numpy_type_map
    ImportError: cannot import name 'numpy_type_map'

我也试过谷歌很多次，但我找不到解决它的方法。 我能做什么？ 我正在使用 PyTorch 0.4.1 和 pytorch nightly 1.0.0-dev。

编辑：感谢 sancelot，我设法解决了这个错误（PyTorch 0.4.0 做到了）。 无论如何，现在我有另一个错误：

    Traceback (most recent call last):
    File "tools/train_net_step.py", line 27, in <module>
    from modeling.model_builder import Generalized_RCNN
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/modeling/model_builder.py", line 11, in <module>
    from model.roi_pooling.functions.roi_pool import RoIPoolFunction
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/model/roi_pooling/functions/roi_pool.py", line 3, in <module>
    from .._ext import roi_pooling
    File "/home/federico/PycharmProjects/Detectron.pytorch/lib/model/roi_pooling/_ext/roi_pooling/__init__.py", line 3, in <module>
    from ._roi_pooling import lib as _lib, ffi as _ffi
    ImportError: /home/federico/PycharmProjects/Detectron.pytorch/lib/model/roi_pooling/_ext/roi_pooling/_roi_pooling.so: undefined symbol: PyInt_FromLong

这次我无法得到的是：这是外部库给出的错误吗？ 我正在使用以前由我的教授制作的 anaconda 环境，他已将其用于 Detectron ......所以我无法猜测为什么我会得到这个。

Answer 1

是的，这是由于 Pytorch 版本不匹配造成的。 解决方案取决于您愿意去的程度，有时如果您可以接受黑客攻击并且只是让它运行，那么只需复制粘贴旧版本的numpy_type_map ：

numpy_type_map = {
     'float64': torch.DoubleTensor,
     'float32': torch.FloatTensor,
     'float16': torch.HalfTensor,
     'int64': torch.LongTensor,
     'int32': torch.IntTensor,
     'int16': torch.ShortTensor,
     'int8': torch.CharTensor,
     'uint8': torch.ByteTensor,
 }

或者，直到1.1.0版本，你可以尝试替换import语句，

来自：

from torch.utils.data.dataloader import numpy_type_map

到：

from torch.utils.data._utils.collate import numpy_type_map

Nb 仍然会在更新的版本中崩溃。 同样，这是一个hacky的快速修复解决方案。

Answer 2

我想检测器和您正在使用的所需 pytorch 版本之间存在版本不匹配。 如果您查看最新的 pytorch 源代码，则没有 numpy_type_map 组件。 https://github.com/pytorch/pytorch/blob/master/torch/utils/data/dataloader.py