weixin_53329734 2021-05-31 11:10 采纳率: 76.8%
浏览 2931
已采纳

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条)

报告相同问题?

悬赏问题

  • ¥30 Matlab打开默认名称带有/的光谱数据
  • ¥50 easyExcel模板 动态单元格合并列
  • ¥15 res.rows如何取值使用
  • ¥15 在odoo17开发环境中,怎么实现库存管理系统,或独立模块设计与AGV小车对接?开发方面应如何设计和开发?请详细解释MES或WMS在与AGV小车对接时需完成的设计和开发
  • ¥15 CSP算法实现EEG特征提取,哪一步错了?
  • ¥15 游戏盾如何溯源服务器真实ip?需要30个字。后面的字是凑数的
  • ¥15 vue3前端取消收藏的不会引用collectId
  • ¥15 delphi7 HMAC_SHA256方式加密
  • ¥15 关于#qt#的问题:我想实现qcustomplot完成坐标轴
  • ¥15 下列c语言代码为何输出了多余的空格