run error in 3090

[JJun-Guo]

does the mosnet can only run in 2080? when running in 3090,many errors happen

[WHUfreeway]

I tried to trained it on cpu,it works; but while it comes to using gpu it occured an InternalError says my video memory use out