Question

我正在尝试使用this回购。但是我遇到了这个错误

File "/home/maq/PycharmProjects/pytorch-retinanet/visualize.py", line 106, in <module>
    main(coco_path, path_to_model)

File "/home/maq/PycharmProjects/pytorch-retinanet/visualize.py", line 53, in main
    retinanet = torch.load(path_to_model)

File "/home/maq/PycharmProjects/pytorch4python2/pytorch-retinanet/local/lib/python2.7/site-packages/torch/serialization.py", line 358, in load
    return _load(f, map_location, pickle_module)

File "/home/maq/PycharmProjects/pytorch4python2/pytorch-retinanet/local/lib/python2.7/site-packages/torch/serialization.py", line 542, in _load
    result = unpickler.load()

File "/home/maq/PycharmProjects/pytorch-retinanet/model.py", line 9, in <module>
    from lib.nms.pth_nms import pth_nms

File "/home/maq/PycharmProjects/pytorch-retinanet/lib/nms/pth_nms.py", line 2, in <module>
    from ._ext import nms

File "/home/maq/PycharmProjects/pytorch-retinanet/lib/nms/_ext/nms/__init__.py", line 3, in <module>
    from ._nms import lib as _lib, ffi as _ffi

ImportError: /home/maq/PycharmProjects/pytorch-retinanet/lib/nms/_ext/nms/_nms.so: undefined symbol: __cudaPopCallConfiguration

我尝试使用此link对它进行故障排除。添加所有路径。在编译repo的python环境中产生结果：

$ python -c "import torch; print(torch.version.cuda)"
9.0.176

$nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176

$  python -c "import torch; print(torch.__version__)"
0.4.1

$python -c "import torch; print(torch.cuda.is_available())"
True

我正在使用Ubuntu 16.04。

PS：终端中的外部环境。除

外，所有命令的结果相同

$  python -c "import torch; print(torch.__version__)"
1.0.1.post2

但是我正在python环境中运行repo，所以我认为这不重要。

如何修复“ /home/......./_nms.so未定义的符号__cudapopcallconfiguration”？

0 个答案: