search

vt-vl-lab / FGVC

[ECCV 2020] Flow-edge Guided Video Completion
Other
1.55k stars 263 forks source link

Questions about the cuda version #1

Closed raywzy closed 3 years ago

raywzy commented 3 years ago

Hi,

Thanks for the great work! I met some problems about the cuda version. For RAFT, the requirement version of cuda is 10.1. However, for environment FGVC, cuda >10 is not supported for pytorch 0.4. It will be a littile incovenient if we need to switch the cuda version during inference.

gaochen315 commented 3 years ago

Sorry for the inconvenience. The current code contains an image inpainting component, which requires Pytorch 0.4.0. I will try to find an image inpainting code that uses a newer Pytorch version.

gaochen315 commented 3 years ago

Please check the latest code. There is only one environment, i.e., Pytorch 1.6.0, CUDA 10.1.

raywzy commented 3 years ago

Please check the latest code. There is only one environment, i.e., Pytorch 1.6.0, CUDA 10.1.

Thanks for your help!