search

sniklaus / pytorch-pwc

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

AttributeError: module 'correlation' has no attribute 'ModuleCorrelation' #12

Closed wine3603 closed 5 years ago

wine3603 commented 5 years ago

Hi man, I am using pythorch with python 3.6 cuda 8.0, when I run

python run.py

I got this error:

Traceback (most recent call last): File "run.py", line 293, in moduleNetwork = Network().cuda().eval() File "run.py", line 268, in init self.moduleTwo = Decoder(2) File "run.py", line 170, in init self.moduleCorrelation = correlation.ModuleCorrelation() AttributeError: module 'correlation' has no attribute 'ModuleCorrelation'


Could you help me ?
sniklaus commented 5 years ago

Have you seen #7?

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

wine3603 commented 5 years ago

Thanks a lot!!! fixed it by installing cupy-cuda80 !

yuanjunchai commented 4 years ago

Yeah, I solve this problem by installing cupy-cuda100(For CUDA 10.0). More details refer to https://cupy.chainer.org/

liubai01 commented 4 years ago

For those who used Ubuntu18, Please use cupy-cuda101(100 may also work). Or you will be faced with weird CompileException in cupy-cuda91 or earlier. Hope someone faced with the same problem will see this issue:).

yuanjunchai commented 4 years ago

For those who used Ubuntu18, Please use cupy-cuda101(100 may also work). Or you will be faced with weird CompileException in cupy-cuda91 or earlier. Hope someone faced with the same problem will see this issue:).

Get it! Thank you!