NVlabs / stylegan3

Official PyTorch implementation of StyleGAN3
Other
6.3k stars 1.11k forks source link

4090 Support? #212

Open makeitrad opened 1 year ago

makeitrad commented 1 year ago

I've been able to get stylegan3 to work using the docker img however it's running PyTorch 1.10 which doesn't work well with the 4090? Will there be any support for this card in the future or PyTorch 1.13 support?

Would seriously love to train models locally with my new card.

Running the 4090 through docker gives me speeds about three times slower than the A100. Should be much closer to the same speed I believe.