weixin_53329734 2021-05-31 11:10 采纳率: 78.2%
浏览 2909
已采纳

RuntimeError: CUDA error: invalid device ordinal

这个是什么原因呢,程序在pycharm上运行,使用的是pytorch框架,然后报错了。

在网上查询时都说是gpu被占用,但是我不知道应该怎么改,

关于device的代码如下

关于gpu的代码如下

怎么改??求解

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2021-05-31 15:52
    关注

    在程序的前面加上,后面的数字要看你的显卡数目,意思是对该程序显示哪几张显卡可以使用。如果只有一张的话,要改成0.然后就是torch设置显卡的问题。

    最好这么写:torch.device('cuda:0')

    import os
    
    os.environ['CUDA_VISIBLE_DEVICES']='0,1,2,3' #4路显卡
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

悬赏问题

  • ¥20 qt中connect两个signal
  • ¥20 pix2pixHD运行测试命令时出现数据类型错误无法反向传播的问题
  • ¥15 python处理Excel符合条件的行自动填写数据分类
  • ¥15 汇编hook举例并讲解(通俗易懂,学习用)
  • ¥20 用c++语言模拟键盘电子琴设计
  • ¥15 STM32cubemx生成keil工程,有问题与正常的情况不同,求解!
  • ¥15 如何自动点击银行app的安全键盘,实现密码自动输入
  • ¥15 关于四边形重叠的问题
  • ¥15 用verilog语言设计一个简易的八音符电子琴,可通过按键输入来控制音响。演奏时可以选择是手演奏(由键盘输入)或自动演奏已存入的乐曲。能够自动演奏多首乐曲,且每首乐曲可重复演奏
  • ¥15 sap gui脚本每次到导出Excel的时候就停住不动。不会另存为。