SenHe / DLFS

Other
66 stars 17 forks source link

Can I use RTX3090 to train this model? #3

Closed yummyJade closed 2 years ago

yummyJade commented 2 years ago

I want to run this model in RTX3090,but the process will dead when it meet netG.cuda(gpu_ids[0]), but I can run this code normally in 2080. I can't figure out what the problem is. Hope for your response.