Pytorch训练问题：AssertionError: Invalid device id

在Linux中使用显卡训练网络时，一般会通过device id来确定使用的显卡。我们从GitHub上获取的源码中的device id和我们本地的device id肯定不一致，所以训练时一定要注意device id修改。

以下示例：

源码：

model = nn.DataParallel(model.cuda(), device_ids=[0,1]

源码中使用了id为0和1 的显卡进行训练。

本地训练报错：

AssertionError: Invalid device id

本地显卡指示：

CUDA Device count: 1

本地只有一个显卡，代码中带入了2个id，这时候肯定会报错。修改代码如下：

model = nn.DataParallel(model.cuda(), device_ids=[0]

注意：

在使用多显卡进行训练时，一定要注意显卡id设置。如遇问题可以参考：

https://blog.csdn.net/qq_41563394/article/details/106555626

https://www.codeleading.com/article/23452065003/

Pytorch训练问题：AssertionError: Invalid device id相关推荐

AssertionError: Invalid device id 和RuntimeError: CUDA error: invalid device ordinal
我在使用torch多卡并行时出现了这个两个问题. ##问题一:AssertionError: Invalid device id,即无效的设备id 出现的原因:结合代码解释: import ...os ...
pytorch测试报错：RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module
模型在服务器多gpu上训练,测试在自己台式机上进行,只有一块gpu,测试报错: File "/home/fuxueping/sdb/PycharmProjects/face_recognit ...
编写transformers的自定义pytorch训练循环（Dataset和DataLoader解析和实例代码）
文章目录一.Dataset和DataLoader加载数据集 1.torch.utils.data 2. 加载数据流程 3. Dataset 4. dataloader类及其参数 5. dataloa ...
让PyTorch训练速度更快，你需要掌握这17种方法
选自efficientdl.com 作者:LORENZ KUHN 机器之心编译编辑:陈萍掌握这 17 种方法,用最省力的方式,加速你的 Pytorch 深度学习训练. 近日,Reddit 上一个帖 ...
PyTorch训练加速17种技巧
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达文自机器之心作者:LORENZ KUHN 编辑:陈萍掌握这 ...
这17 种方法让 PyTorch 训练速度更快！
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:选自 | efficientdl.com 作者 | LO ...
如何用PyTorch训练图像分类器
本文为 AI 研习社编译的技术博客,原标题 : How to Train an Image Classifier in PyTorch and use it to Perform Basic Infe ...
pytorch dali 加速 dali支持的数据处理列表，mxnet tensorflow caff读取数据转换 pytorch训练
使用dali加速,前提是 gpu没有跑满,不然效果也不大 +apex 混合精度训练 5.tf ,mxnet ,caffe, 数据输入,转换成 pytorch 训练 https://github.com ...
浅谈 iOS Device ID 的修改
Dark side of the Force 最近有一篇文章介绍了如何实现 AppStore App 自动下载,笔者看后收获良多.不过文中只介绍了如何去模拟用户的操作来完成下载,并没有涉及抹机.I ...
pytorch训练MNIST
本文记录了pytorch训练MNIST数据集的过程,通过本文可熟悉pytorch训练的大体操作过程. 一.导入各种模块 import torch import torch.nn as nn impor ...

Pytorch训练问题：AssertionError: Invalid device id

Pytorch训练问题：AssertionError: Invalid device id相关推荐

最新文章

热门文章