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

纬领网络

12534人浏览 · 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。

天启AI社区

GitCode 天启AI是一款由 GitCode 团队打造的智能助手，基于先进的LLM（大语言模型）与多智能体 Agent 技术构建，致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话，还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力，真正做到“一句话，让 Al帮你完成复杂任务”。

更多推荐