search

zllrunning / video-object-removal

Just draw a bounding box and you can remove the object you want to remove.
MIT License
2.65k stars 476 forks source link

undefined symbol: __cudaRegisterFatBinaryEnd if run demo.py #31

Open yaroslavmavliutov opened 3 years ago

yaroslavmavliutov commented 3 years ago

Hello! When I try ro run demo.py , i obtain an error

Traceback (most recent call last): File "demo.py", line 3, in from inpaint import inpaint File "/content/video-object-removal/inpaint.py", line 5, in from inpainting.model import generate_model File "/content/video-object-removal/inpainting/model.py", line 3, in from inpainting.models import vinet File "/content/video-object-removal/inpainting/models/vinet.py", line 6, in from inpainting.models.flowmodules import (WarpingLayer, LongFlowNetCorr, MaskEstimator ) File "/content/video-object-removal/inpainting/models/flow_modules.py", line 7, in from inpainting.models.correlation_package.modules.correlation import Correlation File "/content/video-object-removal/inpainting/models/correlation_package/modules/correlation.py", line 3, in from ..functions.correlation import CorrelationFunction File "/content/video-object-removal/inpainting/models/correlation_package/functions/correlation.py", line 3, in from .._ext import correlation File "/content/video-object-removal/inpainting/models/correlation_package/_ext/correlation/init.py", line 3, in from ._correlation import lib as _lib, ffi as _ffi ImportError: /content/video-object-removal/inpainting/models/correlation_package/_ext/correlation/_correlation.so: undefined symbol: __cudaRegisterFatBinaryEnd

yaroslavmavliutov commented 3 years ago

I used pytorch 0.4.0, cuda 8.0 (tried 9.0 also)

dueToLife commented 2 years ago

I met this problem under pytorch 0.4.0, python 3.5.6, cuda 11.1, gcc 7.6 and solved it by changing environment to pytorch 0.4.0, python 3.5.6, cuda 9.0, gcc 5.5.0. After changing env, rerun .sh files to compile dependencies. Hope it useful.

overwhelmedxh commented 2 years ago

I met this problem under pytorch 0.4.0, python 3.5.6, cuda 11.1, gcc 7.6 and solved it by changing environment to pytorch 0.4.0, python 3.5.6, cuda 9.0, gcc 5.5.0. After changing env, rerun .sh files to compile dependencies. Hope it useful. that is useful, gcc 5.4 also works!