search

junfu1115 / DANet

Dual Attention Network for Scene Segmentation (CVPR2019)
MIT License
2.41k stars 483 forks source link

undefined symbol: __cudaRegisterFatBinaryEnd #115

Open Stephanessy opened 4 years ago

Stephanessy commented 4 years ago

My env: cuda 8.0, pytorch 0.4.1

Traceback (most recent call last):
  File "train.py", line 18, in <module>
    import encoding.utils as utils
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/__init__.py", line 13, in <module>
    from . import nn, functions, dilated, parallel, utils, models, datasets
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/nn/__init__.py", line 12, in <module>
    from .encoding import *
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/nn/encoding.py", line 18, in <module>
    from ..functions import scaledL2, aggregate, pairwise_cosine
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/functions/__init__.py", line 2, in <module>
    from .encoding import *
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/functions/encoding.py", line 14, in <module>
    from .. import lib
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/lib/__init__.py", line 20, in <module>
    ], build_directory=gpu_path, verbose=False)
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 514, in load
    with_cuda=with_cuda)
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 690, in _jit_compile
    return _import_module_from_library(name, build_directory)
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 773, in _import_module_from_library
    return imp.load_module(module_name, file, path, description)
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/mmvc/anaconda3/envs/torch041/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: /home/mmvc/anaconda3/envs/torch041/lib/python3.6/site-packages/encoding/lib/gpu/enclib_gpu.cpython-36m-x86_64-linux-gnu.so: undefined symbol: __cudaRegisterFatBinaryEnd

How to fix this?