pytorch 使用指定的GPU RuntimeError: CUDA error: invalid device ordinal

城俊BLOG

已于 2023-12-10 15:50:27 修改

阅读量3.4w

点赞数 33

CC 4.0 BY-SA版权

分类专栏： Pytorch # DL-报错 Python 文章标签： pytorch 人工智能 python

于 2020-07-31 20:15:50 首次发布

本文链接：https://blog.csdn.net/qxqxqzzz/article/details/107720675

Python 同时被 3 个专栏收录

307 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

DL-报错

122 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

Pytorch

47 篇文章

订阅专栏

当使用PyTorch指定GPU时，可能会遇到`CUDA error: invalid device ordinal`的错误。错误可能源于逗号误用、分布式代码中GPU数量与指定序号不匹配、多处设置GPU使用不一致、环境变量与`torch.cuda.set_device`冲突等问题。解决方案包括正确设置`CUDA_VISIBLE_DEVICES`，协调代码中GPU使用的一致性，以及理解`os.environ['CUDA_VISIBLE_DEVICES']`与`torch.cuda.set_device`的关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pytorch使用指定GPU报错：

Traceback (most recent call last):
  File "test_bed/process_deepglint.py", line 102, in <module>
    pred_dataset(outputFile)
  File "test_bed/process_deepglint.py", line 36, in pred_dataset
    pred_loader_deepg, model, criterion, attrWeights, useArcface = main()
  File "/home/user1/main_cs_0708.py", line 114, in main
    model = models.__dict__[arch]()
  File "/home/user1/models/arc_face.py", line 35, in arcface
    learner = arc_face.face_learner(conf, inference=True)
  File "/home/user1/arc_face/Learner.py", line 24, in __init__
    self.model = Backbone(conf.net_depth, conf.drop_ratio, conf.net_mode).to(conf.device)
  File "/home/user1/miniconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 386, in to
    return self._apply(c

了解本专栏

超级会员免费看