video-object-removal
video-object-removal copied to clipboard
Error when running bash install.sh
Hello everyone, i just got the following error when running bash install.sh:
Compiling correlation kernels by nvcc...
Traceback (most recent call last):
File "build.py", line 3, in
Is anyone can help me?
My PC is Ubuntu 18.04 LTS, Python 3.6, Pytorch 1.4, CUDA 10.1, GTX1070 GPU.
change
from torch.utils.ffi import create_extension
to
from torch.utils.cpp_extension import BuildExtension
and change
ffi = create_extension(...)
to
ffi = BuildExtension(...)
I used to meet this also. It's because your nvcc can't find THC.h and THCGeneric.h. see this line and this line. Check if it is a true path of pytorch:
print(os.path.dirname(torch.__file__))
And due to your high pytorch version,some api was deprecated,use their substitute in pytorch 1.0+
Thank all bros. Let me do.