Как исправить "/home/......./_nms.so неопределенный символ __cudapopcallconfiguration"? - PullRequest
Новая
       46

Как исправить "/home/......./_nms.so неопределенный символ __cudapopcallconfiguration"?

0 голосов
/ 29 мая 2019

Я пытаюсь использовать это репо. Но я получаю эту ошибку 

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

Я попытался устранить неполадку, используя эту ссылку . Все пути добавлены. Результат в среде 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, поэтому я не думаю, что это должно иметь значение.

...