Can you give the specific environment configuration, including the versions of cuda, pytorch and various libraries?

[swjtuer0762]

I solved the problem。If you want to ask me how to do, please comment, I will teach you.