search

chigozienri / VQGAN-CLIP-animations

134 stars 28 forks source link

A100 Support #29

Closed rs3dtouch closed 2 years ago

rs3dtouch commented 3 years ago

When using a A100 (Pro+) the processing is dramatically slow, several times slower than a P100/V100.

This is probably an issue with one of the libraries, but I could not figure out which one.

Any comments appreciated, thank you very much.

chigozienri commented 3 years ago

Thanks for the report. This is quite difficult for me to test myself, as I don't have Colab Pro+. I could try to do a troubleshooting call with you, if that works?

chigozienri commented 2 years ago

@rs3dtouch or @tomfpritchard I'd love to fix this, if you are available at any point to do some tests with me

chigozienri commented 2 years ago

For now, I've added a recommendation to reload until you get a V100 or P100, and I'm closing this issue. If I get an opportunity to test on a Colab Pro account I can re-open it.

chigozienri commented 2 years ago

@rs3dtouch or @tomfpritchard I might have a fix, could you try running torch.backends.cudnn.enabled = False after the 'installation and loading of libraries' step?

chigozienri commented 2 years ago

I've pushed a possible fix, but have no way to test. If I get more reports, I'll re-open the issue.