Open ltkong218 opened 5 years ago
I've ever got this error before when using PyTorch-1.1.0. I suggest you to run with PyTorch-1.0.0 and Python-3.7.
@ltkong218 How do you solve this error? I met the same error even though I ran the code with PyTorch-1.0.0 and Python-3.7.
Use CUDA-9.2 and run
python setup.py build
python setup.py install
under ./models/networks/correlation_package and ./models/networks/resample2d_package respectively, then I can run the code.
I replace the correlation module with that in FlowNet2 https://github.com/NVIDIA/flownet2-pytorch
@ltkong218 Thanks for your reply.
Hello, I tried this code with cuda-9.0 and cuda-10.0 with pytorch-1.1.0, and I have built and installed the correlation module. But when I run
python train.py
, it gets an error above, should I use cuda-9.2?