AttributeError module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘

在安装Pytorch时如果cpu版本的，不是gpu版本的。会提示AttributeError module 'torch._C' has no attribute '_cuda_setDevice'，所以，需要在python命令后面加上--gpu_ids -1，问题解决。运行如下命令，问题就解决了。python train.py --data_dir sample_dataset--gpu_ids

纬领网络

12580人浏览 · 2021-07-01 16:43:31

纬领网络 · 2021-07-01 16:43:31 发布

在安装Pytorch时如果cpu版本的，不是gpu版本的。会提示AttributeError module 'torch._C' has no attribute '_cuda_setDevice'，所以，需要在python命令后面加上 --gpu_ids -1，问题解决。

运行如下命令，问题就解决了。

python train.py --data_dir sample_dataset --gpu_ids -1

或者，也可以在文件中，直接把参数gpu_ids的默认值改成-1。

FlagOS智算系统软件栈

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

更多推荐

不要死磕CUDA，国内首个Triton技术大会官宣，AI芯片编程迎来新范式

你将听到 Triton 诞生背后的第一性原理与未来演进的宏伟蓝图，获取关于 AI 编译器的最新学术研究报告，了解众智 FlagOS 社区生态的最新进展，更有对增强版 Triton 编译器特性的深度讲解。作为国内专注于 Triton 技术及其生态的开发者盛会，我们诚挚地邀请全球的 AI 研究员、高性能计算工程师、编译器专家、编程语言爱好者，以及所有希望突破 AI 芯片编程复杂性的开发者，与我们一起，