其实还是numpy和pytorch版本不匹配的问题，建议可以用bard或者new bing搜一下对应的版本，我没有找到相应的文档，问了bard给出的答案。