在安装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。
AttributeError module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘
最新推荐文章于 2024-10-30 10:46:08 发布
本文介绍了解决在使用PyTorch CPU版本时遇到的AttributeError问题的方法。通过在命令行中添加特定参数或修改默认配置，可以避免与GPU相关属性引发的错误。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
4585
到【灌水乐园】发言
