原因:pytorch没有配置GPU/CUDA(若已经安装并配置GPU+CUDA)
到pytorch官网安装适合的版本:
https://pytorch.org/get-started/locally/
在这里插入图片描述
将最底层的 Run this Command复制,到pycharm的终端运行。

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐