Some trouble when training EPNet on 3090

[quliang93]

Hi, guys
I tried to train EPNet on 3090, I did the same thing just as the author told, and the environment is right.
But I failed to train, the training process will shutdown.
Oh, I do this on my 3090, is there a problem?

Could anybody give some advice?
Thanks a lot.

[faziii0]

try with cudatoolkit 11.3 pyhton 3.8