Update the pytorch version in repository version to compatible with newer GPU

hmorimitsu / ptlflow

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.

Apache License 2.0

250 stars 33 forks source link

Update the pytorch version in repository version to compatible with newer GPU #41

Closed TranNg65 closed 1 year ago

TranNg65 commented 1 year ago

Hi hmorimitsu,

I got this warning when using the repository with RTX 3090:

"NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70."

Can you update the codes for it to be compatible with newer GPUs?

Thank you for your good works.

TranNg65 commented 1 year ago

the issue is solved after reinstalling pytorch.