device = torch.device("cuda:{}".format(args.gpu) if torch.cuda.is_available() else "cpu")
RuntimeError: Invalid device string: 'cuda:-1'
请问这种情况如何解决？
RuntimeError: Invalid device string: 'cuda:-1'
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
分享- 邀请回答
-
2条回答 默认 最新
元气少女缘结神 2023-02-06 18:30关注试下改成
device = torch.device('cuda' if torch.cuda.is_available() and use_cuda else 'cpu')解决 无用评论 打赏举报 分享
- 2021-05-31 11:10回答 2 已采纳 在程序的前面加上，后面的数字要看你的显卡数目，意思是对该程序显示哪几张显卡可以使用。如果只有一张的话，要改成0.然后就是torch设置显卡的问题。 最好这么写：torch.device('cuda:
- 2020-09-05 22:14回答 1 已采纳 检查显示卡是否兼容、驱动程序、cuda sdk和cudaa 的安装。
- 2020-09-24 19:26回答 1 已采纳 导致这个错误的原因是内存为null，内存没有初始化，具体看下报错的行
- 2024-10-07 19:22无沃的博客 Using the option--gpu=cuda:0instead of--gpu=0 Or update the codetorch.cuda.set_device(args.gpu)withtorch.cuda.set_device(int(args.gpu))
- 2019-04-27 23:25回答 6 已采纳 https://github.com/pytorch/pytorch/issues/20990 链接答案来源 import torch torch.cuda.current_device()
- 2023-04-04 09:44回答 4 已采纳 该回答通过自己思路及引用到各个渠道搜索综合及思考,得到内容具体如下： 根据您提供的信息，这个错误是由于GPU内存不足引起的。在训练yolov7标准版时，需要更多的GPU内存，而您的GPU内存已经被占用
- 回答 2 已采纳 这几个部分你用int()包裹起来试试
- 2021-10-31 10:48莫余的博客 device = torch.device('cuda：0' if torch.cuda.is_available() and use_cuda else 'cpu') 应修改为： os.environ['CUDA_VISIBLE_DEVICES'] = '0' device = torch.device('cuda' if torch.cuda.is_available() and...
- 2023-03-15 10:55回答 3 已采纳 该回答引用GPTᴼᴾᴱᴺᴬᴵ这个错误提示表明你的 PyTorch 模块是使用 API 版本 0x10 编译的，但是当前安装的 NumPy 版本的 API 版本是 0xf。这通常是由于 NumPy 版本
- 2023-04-07 10:43回答 1 已采纳 这样试试呢？ batch_indices = torch.arange(B, dtype=torch.long).to(device).view(B, 1).repeat(repeat_shape)
- 2022-01-21 15:04回答 1 已采纳 维度不匹配。他希望你该层的输入为[32, 512, 38, 38]，32位batch size，512为通道数，后面两位是H和W，但是你的卷积之后的通道数在该层只有64通道，16 batch size
- 丢掉幻想,准备斗争的博客 RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors might be 原因： 当前环境的显卡 与程序cuda: gpu_idx设定的显卡不匹配引起 解决方案： 添加 CUDA_VISIBLE_DEVICES=0 原来代码 python your_...
- 2022-09-19 20:02回答 5 已采纳 请问这里面的数据是怎样的 ./data/products/ogbn_products/mapping/labelidx2productcategory.csv.gz我跑跑看，光这样子暂时排查不出什么问
- 2024-10-21 20:52Kyov的博客 device = torch.device("cuda:1") # 这应该改为 cuda:0。然后将对应文件中的gpu数量改为0（gpu都是从0开始编号的）查看此设备有多少GPU可用（我的是只有一个）我遇到的问题是，GPU数量不够，只有一个。在IDE中可用...
- 2022-05-23 16:22Jane_JU的博客 开头加上 import os os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘0,1,2,3’ 改为： device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’ )
- 没有解决我的问题, 去提问
问题事件
已结题
（查看结题原因） 2月6日-
创建了问题
2月6日
悬赏问题
- ¥20 如何让程序ab.eXe自已删除干净硬盘里的本文件自己的ab.eXe文件
- ¥50 爬虫预算充足，跪巨佬
- ¥15 滑块验证码拖动问题悬赏
- ¥15 Wanted but not invoked:Actually, there were zero interactions with this moc
- ¥20 怎么驱动tb6612
- ¥15 Arcgis 3D效果点要素
- ¥15 在执行yolo训练脚本的时候，没有报错，在扫描val这一步后就推出执行了
- ¥15 delphi开发的WEBSERVER改用HTTPS协议
- ¥15 pic16f877A单片机的计数器proteus仿真失效
- ¥100 调查 Vitis AI 中验证 .xmodel 量化后的正确性