Does the project support cuda11 and RTX3090?

songweige / TATS

Official PyTorch implementation of TATS: A Long Video Generation Framework with Time-Agnostic VQGAN and Time-Sensitive Transformer (ECCV 2022)

MIT License

267 stars 17 forks source link

Does the project support cuda11 and RTX3090? #3

Closed yrcong closed 2 years ago

yrcong commented 2 years ago

Hi,

have you tried to train your model on a RTX30 GPU?

songweige commented 2 years ago

Hi, @yrcong, thank you for your interests! We haven't tried to to train the model on a RTX30 gpu. But I did once use it with cuda11. I don't think there will be an issue that prevents the training on such gpus. You do need to choose the correct pytorch and pytorch-lightning versions. Feel free to let me know if you run into any issues!