roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 565 forks source link

torch.utils.ffi is deprecated with pytorch 0.4.1 #202

Closed matchumen closed 5 years ago

matchumen commented 5 years ago

When I run tools/test_net.py, I end up with ImportError even though I'm using PyTorch 0.4.1.

Traceback (most recent call last): File "tools/test_net.py", line 14, in from core.test_engine import run_inference # tady to padá pro pána jána File "/home/matchumen/quest/detectors/mask-rcnn.pytorch/lib/core/test_engine.py", line 39, in from modeling import model_builder File "/home/matchumen/quest/detectors/mask-rcnn.pytorch/lib/modeling/model_builder.py", line 11, in from model.roi_pooling.functions.roi_pool import RoIPoolFunction File "/home/matchumen/quest/detectors/mask-rcnn.pytorch/lib/model/roi_pooling/functions/roi_pool.py", line 3, in from .._ext import roi_pooling File "/home/matchumen/quest/detectors/mask-rcnn.pytorch/lib/model/roi_pooling/_ext/roi_pooling/init.py", line 2, in from torch.utils.ffi import _wrap_function File "/home/matchumen/.local/lib/python3.6/site-packages/torch/utils/ffi/init.py", line 1, in raise ImportError("torch.utils.ffi is deprecated. Please use cpp extensions instead.") ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.

System information

matchumen commented 5 years ago

My fault. Python executable pointed to '.local' directory, which contained PyTorch 1.0 for some reason.