Updated resample2d.py (adding simple import)

NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Other

3.09k stars 739 forks source link

Updated resample2d.py (adding simple import) #217

Closed hugothimonier closed 9 months ago

hugothimonier commented 4 years ago

Adding simple 'import torch' before calling 'import resample2d_cuda' allows to avoid errors in Pytorch 1.1 + Cuda 10 + Python 3.6. env of the type :

ImportError: local/lib/python3.6/site-packages/resample2d_cuda-0.0.0-py3.6-linux-x86_64.egg/resample2d_cuda.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at5ErrorC1ENS_14SourceLocationESs

and possibly other env with Pytorch 0.4.1