search

sniklaus / pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
GNU General Public License v3.0
616 stars 123 forks source link

module 'correlation' has no attribute 'ModuleCorrelation' #7

Closed taylover-pei closed 5 years ago

taylover-pei commented 5 years ago

When I try to run your code, errors happened like this: image I uese python3.6, pytorch0.4。So, what should I do to solve the problem. Looking forward to your reply. Thank you very much.

sniklaus commented 5 years ago

As a quick sanity check, could you run python --version and let me know what it prints?

amjltc295 commented 5 years ago

I got the same error as well. It seems to happen when cupy is not installed correctly. I solved it by using pip install cupy-cuda90 to install cupy (follow the instruction of https://github.com/cupy/cupy), with Python 3.6.6 and Pytorch 0.4.1.