Pytorch"填坑“记（长更）

最新推荐文章于 2022-07-11 20:07:34 发布
原创 最新推荐文章于 2022-07-11 20:07:34 发布 · 563 阅读 · 0 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#pytorch #GPU

本文解决了一个常见的PyTorch训练过程中遇到的问题：当指定GPU时出现的RuntimeError。问题源于GPU编号不一致，解决方法是通过设置环境变量CUDA_VISIBLE_DEVICES来调整可见的GPU编号，并确保程序中指定的GPU编号是从0开始。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

问题：

存在多个ＧＰＵ时，指定GPU时的错误：RuntimeError: cuda runtime error (10) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:34


解决方法之一：

CUDA_VISIBLE_DEVICES=6（根据自己的ＧＰＵ个数，任意改变）
./train.py --gpu 0(不要改变它！！！从０开始）

原因之一：

因PyTorch程序设置的gpu_id编号未从0开始导致

pytorch 使用指定的GPU RuntimeError: CUDA error: invalid device ordinal
Hello Word!
07-31 3万+
pytorch使用指定GPU报错： Traceback (most recent call last): File "test_bed/process_deepglint.py", line 102, in <module> pred_dataset(outputFile) File "test_bed/process_deepglint.py", line 36, in pred_dataset pred_loader_deepg, model, criterion,
PyTorch~cpp_extension
whaosoft143ai的博客
12-19 650
还是pytorch哈~~ 结合 Python 与 C++ 各自的优点，在 PyTorch 中加入 C++ / CUDA的扩展，详细解释C++/CUDA 算子实现和调用全流程本期主题：结合 Python 与 C++ 各自的优点，在 PyTorch 中加入 C++ / CUDA的扩展，而让我们自己更好地使用工具而不为工具所束缚。代码来源：MMCV, PyTorch。注：C++ / CUDA 扩展一般有”预编译“ 与 ”实时编译“ （just-in-time, JIT）模式。本期主要介绍”预编译“模式。
PyTorch 1.0 中文文档：torch.utils.cpp_extension
weixin_30469895的博客
03-07 2024
译者: belonHan torch.utils.cpp_extension.CppExtension(name, sources, *args, **kwargs) 创建一个C++的setuptools.Extension。 便捷地创建一个setuptools.Extension具有最小（但通常是足够）的参数来构建C++扩展的方法。 所有参数都被转发给setuptools.Extension...
pytorch拓展-cpp extension
CV老李的博客
06-09 1472
pytorch的cpp拓展
【Bugs】RuntimeError CUDA out of memory
SUFEHeisenberg的博客
12-26 1183
【Bugs】RuntimeError: CUDA out of memory. 报错如下： Traceback (most recent call last): File "xxx.py", line 110, in <module> loss.backward() File "/nfsshare/apps/anaconda3/lib/python3.7/site-packages/torch/tensor.py", line 185, in backward torch
pytorch中为Module和Tensor指定GPU
VisualCortex
06-02 7758
pytorch指定GPU 在用pytorch写CNN的时候，发现一运行程序就卡住，然后cpu占用率100%，nvidia-smi 查看显卡发现并没有使用GPU。所以考虑将模型和输入数据及标签指定到gpu上。 pytorch中的Tensor和Module可以指定gpu运行，并且可以指定在哪一块gpu上运行，方法非常简单，就是直接调用Tensor类和Module类中的 .cuda() 方法。 i...
Pytorch版Faster R-CNN配置运行填坑记录
Watermelon1986的博客
11-28 996
Pytorch Faster R-CNN配置运行填坑记录 之前一直在用Nvidia GTX1080Ti做实验，用到的代码及整套开发环境的版本都较老（Python2.7+Pytorch0.4.0+CUDA8.0），由于20系列显卡或30系列显卡都不再支持CUDA8.0，所以需要进行一次整体升级，在此做一个升级配置环境的填坑记录。
Pytorch的坑
qq_41879767的博客
03-20 551
关于Anaconda环境下的Pytorch安装的TipsAnaconda的安装怎么安装AnacondaPytorch的安装让我们一起来填坑吧 Anaconda的安装 整个戏剧故事的开始自然是要从Anaconda开始了 不过我在安装pytorch的时候其实老早就已经安装好anaconda的环境了 但是作为故事发展需要还是有必要提一下这一点的 ...
win10中pytorch以及torchvision安装填坑
smilexuxu123的博客
07-04 1149
安装pytorch以及torchvision过程中遇到了一些问题，记录下来以便日后复习，也希望给遇到相同问题的小伙伴一些启发。 1.安装pytorch anaconda和cuda的安装过程这里不再说明，本文使用的是python3.6和cuda8.0 参考网上一些大佬的教程用conda命令执行官网上的相关命令时，安装过程实在是太慢了，使用了清华源也不能很好地得到解决。于是转战pip3；使用py...
学习pytorch的小记录
qq_29320285的博客
03-12 278
学习pytorch的小记录原教程地址1.PyTorch神经网络简介1.1 计算梯度1.2 这一部分的可运行代码贴在下面2.训练一个分类器2.1 如何进行数据处理2.2 训练的步骤2.3 数据处理目前暂时后面的还没看完~ 原教程地址 PyTorch简明教程 1.PyTorch神经网络简介 1.1 计算梯度 那里是有一点不太明白的记录一下。 在调用loss.backward()之前，我们需要清除掉te...
PaddleOCR填坑记
Dorthyn的博客
08-12 1374
PaddleOCR部署
RuntimeError: CUDA error (10): invalid device ordinal
小李的研究生学习日记
07-11 5914
在 Pytorch 读取参数时，报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。造成这个错误的原因主要是本地只有一个 GPU (GPU:0)，而程序中使用 GPUs:1。因此，在程序中找到定义 device 的地方，我的如下：将其中 "cuda:1" 修改为 "cuda:0"，程序将不在报错。或者你的配置信息在yaml中：修改对应部分即可......
Pytorch代码遇到THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405/ error=11
weixin_43839245的博客
07-22 4429
在运行pix2pix时遇到了上面的问题： 配置：cuda10.0 cudnn7.4.2 显卡RTX2080ti 安装pytorch(利用pip安装1.1.0,1.0.1,1.5.0都安装了一遍,且相对应的torchvision版本也安装了一遍)都会出现以上问题。 原因是显卡用的RTX 2080Ti，CUDA就要装10以上，这个时候pytorch不能直接用pip装，按照下面指令在终端里面安装。 pip3 install https://download.pytorch.org/whl/cu100/torch-
CUDA error: out of memory (allocate at /pytorch/aten/src/THC/THCCarchingAllocator.cpp:510)
Neptunerr的博客
05-30 1228
在pytorch中，使用多GPU训练，但测试加载模型时使用单张GPU会出现这类报错 解决： 整个model打包保存的情况下，加载模型时应： model_path = 'model.tar' checkpoint = torch.load(model_path) model = checkpoint['model'] if isintance(model, torvh.nn.DataPa...
pytorch模型提示超出内存cuda runtime error(2): out of memory
热门推荐
学习、记录、分享
11-23 2万+
看到这个提示，表示您的GPU内存不足。由于我们经常在PyTorch中处理大量数据，因此很小的错误可能会迅速导致程序耗尽所有GPU; 好的事，这些情况下的修复通常很简单。这里有几个常见检查事项包括： 一、不要在循环训练中累积历史记录。 默认情况下，涉及需要求导/梯度gradients变量的计算将保存在内存中。计算中避免使用这些变量，例如在跟踪统计数据时，这些变量在循环训练中将超出你内存。相反，您...
python出现THCudaCheck FAIL file=/pytorch/aten/src/THC/ThCGeneral.cpp line=405 error=11 : 情况
weixin_39467358的博客
03-15 2899
pip install https://download.pytorch.org/whl/cu110/torch-1.7.0%2Bcu110-cp37-cp37m-linux_x86_64.whl pip install torchvision
损失函数Target size must be the same as input size
WJ_MeiMei的博客
04-01 1万+
出错程序 criterion_modality = torch.nn.BCEWithLogitsLoss() label = Variable(label.cuda()) loss = criterion_modality(outRGB, label) # 出错行 问题 File "C:\Users\Rain\AppData\Local\Programs\Python\Anaconda.3....
PyTorch实现长短期记忆网络(LSTM)指南
在深度学习领域，长短期记忆网络（LSTM）是一种特殊类型的循环神经网络（RNN），非常适合处理和预测时间序列数据中的重要事件，并具有较长的间隔和延迟。PyTorch是一种基于Python的科学计算包，主要用于机器学习和...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值