直接换pytorch版本
官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。
官网
https://pytorch.org/get-started/previous-versions/
直接拷代码下载

Logo

一站式 AI 云服务平台

更多推荐