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

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

报告相同问题？

悬赏问题

  • ¥15 在若依框架下实现人脸识别
  • ¥15 网络科学导论，网络控制
  • ¥100 安卓tv程序连接SQLSERVER2008问题
  • ¥15 利用Sentinel-2和Landsat8做一个水库的长时序NDVI的对比，为什么Snetinel-2计算的结果最小值特别小，而Lansat8就很平均
  • ¥15 metadata提取的PDF元数据，如何转换为一个Excel
  • ¥15 关于arduino编程toCharArray()函数的使用
  • ¥100 vc++混合CEF采用CLR方式编译报错
  • ¥15 coze 的插件输入飞书多维表格 app_token 后一直显示错误，如何解决？
  • ¥15 vite+vue3+plyr播放本地public文件夹下视频无法加载
  • ¥15 c#逐行读取txt文本，但是每一行里面数据之间空格数量不同