如何解决module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘

在运行后面添加--gpu_ids -1如：python train.py--gpu_ids -1原因：因为我们安装的pytorch是CPU版本的

强哥的小迷妹

7690人浏览 · 2021-09-25 10:38:47

强哥的小迷妹  ·  2021-09-25 10:38:47 发布

在运行后面添加--gpu_ids -1

如：python train.py --gpu_ids -1

原因：因为我们安装的pytorch是CPU版本的

# pytorch # 深度学习

华为云开发者联盟

为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。

加入社区

更多推荐

• · 通过HPA+CronHPA组合应对业务复杂弹性伸缩场景
• · FT-FMEA融合混沌演练，零售运营系统韧性架构在线验证实践
• · 如何使用Python和Plotly绘制3D图形

通过HPA+CronHPA组合应对业务复杂弹性伸缩场景

华为云开发者联盟

FT-FMEA融合混沌演练，零售运营系统韧性架构在线验证实践

华为云开发者联盟

如何使用Python和Plotly绘制3D图形

华为云开发者联盟

欢迎加入社区

取消 确定