Import Error

[the-butterfly]

Success build with ubuntu 16.04, cuda 10 and gcc 7.4.
But import error encountered:

In [1]: import torch

In [2]: from box_convolution import BoxConv2d

---

ImportError                               Traceback (most recent call last)
<ipython-input-2-2424917dbf01> in <module>()
----> 1 from box_convolution import BoxConv2d

~/Software/pkgs/box-convolutions/box_convolution/__init__.py in <module>()
----> 1 from .box_convolution_module import BoxConv2d

~/Software/pkgs/box-convolutions/box_convolution/box_convolution_module.py in <module>()
      2 import random
      3 
----> 4 from .box_convolution_function import BoxConvolutionFunction, reparametrize
      5 import box_convolution_cpp_cuda as cpp_cuda
      6 

~/Software/pkgs/box-convolutions/box_convolution/box_convolution_function.py in <module>()
      1 import torch
      2 
----> 3 import box_convolution_cpp_cuda as cpp_cuda
      4 
      5 def reparametrize(

ImportError: /usr/Software/anaconda3/lib/python3.6/site-packages/box_convolution_cpp_cuda.cpython-36m-x86_64-linux-gnu.so: undefined symbol: __cudaPopCallConfiguration

@shrubb

[shrubb]

Oh, Anaconda often doesn't make life easier :(

Most likely, you have two different CUDA toolkits (maybe one in Anaconda and one system-wide) and your PyTorch is using one version while setup.py picks up another.

Check your PyTorch's CUDA version via torch.version.cuda. Then, before build, set CUDA_HOME environment variable to the location of that toolkit. Example:

rm -rf build/
CUDA_HOME=/usr/local/cuda-9.2 python3 -m pip install .

By the way, if you use the CUDA toolkit that is shipped by Anaconda, keep in mind that it is not a real toolkit but some truncated version of it, which is unsuitable for compiling. That might also be the cause.

[the-butterfly]

Thanks for your meticulous reply!
I have just re-install pytorch 1.0.1 with cuda 10, and make install again, everything works fine.

[Flock1]

@the-butterfly, can you tell me if you installed pytorch from source or by using pip? Is your system's CUDA version different from the one installed with torch?

[the-butterfly]

I think system's CUDA version should match with torch.
And my pytorch is installed by pip.

[Flock1]

Okay, thanks