can't run correlation package on google colab

visinf / irr

Iterative Residual Refinement for Joint Optical Flow and Occlusion Estimation (CVPR 2019)

Apache License 2.0

194 stars 32 forks source link

can't run correlation package on google colab #29

Closed RokiaAbdeen closed 4 years ago

RokiaAbdeen commented 4 years ago

I have tried to run the code on google colab using python 3.6, cuda 8.0 and pytorch 0.4.1 but get many errors related to torchvision and I have tried again to run it with newer versions of cuda (9.0 and 10) and pytorch (1.1) but got this error (no package called correlation_cuda) do you have any solution for correlation package problem please ?!

hurjunhwa commented 4 years ago

Hi, Have you successfully installed the correlation package by running this?: https://github.com/visinf/irr/blob/master/install.sh

RokiaAbdeen commented 4 years ago

it works now with torch==1.1.0 and torchvision==0.3.0 thanks alot

hurjunhwa commented 4 years ago

Thanks for sharing!!