Closed fabiopk closed 5 years ago
I am not sure I am afraid, your environment looks good in general. This issue came up before and was resolved by installing cupy
(probably cupy-cuda91
in your case) but I do not understand how that could be the issue that is causing this behavior: https://github.com/sniklaus/pytorch-pwc/issues/7, https://github.com/sniklaus/pytorch-pwc/issues/12. What does python --version
say?
Ok, I think I got it to work.
I used pip install cupy-cuda90
instead of 91 before.
Uninstalling cupy and reinstalling with pip install cupy-cuda91
seemed to fix the issue.
Thanks!
I already did what @fabiopk said but I am still getting this error AttributeError: module 'correlation' has no attribute 'FunctionCorrelation'
Trying to run:
python run.py --model default --first ./images/first.png --second ./images/second.png --out ./out.flo
Gives me the following output:
Using Ubunutu 18.10 Pytorch 1.0.1 CUDA 9.1 Python 3.7.2 GCC 7.3.0
Any ideias?