NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
Other
3.12k stars 739 forks source link

cuda version #287

Closed javlonbek97 closed 6 months ago

javlonbek97 commented 6 months ago

I need help to implement this code but I have been failing for a long time. What parts of the code should be changed in order to be compatible with this: Cuda compilation tools, release 12.1, V12.1.66

print(torch.version) 2.0.0+cu117

NVIDIA-SMI 530.30.02 Driver Version: 530.30.02 CUDA Version: 12.1

stha-prashant commented 5 months ago

I need help to implement this code but I have been failing for a long time. What parts of the code should be changed in order to be compatible with this: Cuda compilation tools, release 12.1, V12.1.66

print(torch.version) 2.0.0+cu117

NVIDIA-SMI 530.30.02 Driver Version: 530.30.02 CUDA Version: 12.1

hi how did you get it to work?

sabitarajbanshi commented 5 months ago

I need help to implement this code but I have been failing for a long time. What parts of the code should be changed in order to be compatible with this: Cuda compilation tools, release 12.1, V12.1.66

print(torch.version) 2.0.0+cu117

NVIDIA-SMI 530.30.02 Driver Version: 530.30.02 CUDA Version: 12.1

hi, I too have been stuck in this for long time. Did you get it work? Please tell me how?