weixin_53329734
weixin_53329734
2021-05-31 11:10
采纳率: 51.9%
浏览 43

RuntimeError: CUDA error: invalid device ordinal

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

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

关于device的代码如下

关于gpu的代码如下

怎么改??求解

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

2条回答 默认 最新

  • qq_34124780
    爱晚乏客游 2021-05-31 15:52
    已采纳

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

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

    import os
    
    os.environ['CUDA_VISIBLE_DEVICES']='0,1,2,3' #4路显卡
    点赞 评论
  • weixin_45393898
    AlanMax 2021-05-31 15:16

    建议这么写: net.cuda()

    点赞 评论

相关推荐