Do you have specific operating environment requirements?

[xhangHU]

Since you only provided pytorch and python versions, but did not export the environment requirement file, I don't know your environment version, can you export an environment requirement file?