TensorFlow: CUDA_ERROR_OUT_OF_MEMORY

最新推荐文章于 2023-09-08 16:23:33 发布
原创 最新推荐文章于 2023-09-08 16:23:33 发布 · 517 阅读 · 1 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#tensorflow #cuda_error #out_of_memory #深度学习 #机器学习

当使用他人已训练的TensorFlow模型遇到CUDA_ERROR_OUT_OF_MEMORY错误时，表明GPU内存不足。即使不希望使用GPU，该错误仍会出现。本文介绍如何通过设置环境变量来禁用GPU，从而避免此问题。

在运行别人已经训练好的模型时报了这个错

TensorFlow: CUDA_ERROR_OUT_OF_MEMORY

通过字面意思也知道是内存不够了，而CUDA指的是GPU。而我使用别人的模型，并不希望使用到GPU。为解决这个问题需要进行如下设置：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
tensorflow CUDA out of memory
无奈的小心酸的博客
05-27 4万+
今天刚把服务器搭建起来 结果程序就跑不起来 当时差点把自己吓尿了 错误类型：CUDA_ERROE_OUT_OF_MEMORY E tensorflow/stream_executor/cuda/cuda_driver.cc:924] failed to alloc 17179869184 bytes on host: CUDA_ERROR_OUT_OF_MEMORY W ./tenso...
FFmpeg CUDA_ERROR_OUT_OF_MEMORY: out of memory
asdasfdgdhh的博客
07-05 809
123
CUDA_ERROR_OUT_OF_MEMORY: out of memory； total memory reported: 34058272768
bashan1604的博客
10-13 1613
环境 Ubuntu: 16.04 CUDA: 10.0 CUDNN: 7.6.5 执行情况 $ python3 train.py {'train': {'data_loader': {'batch_size': 1, 'resize_shape': [600, 600], 'train_and_val': 0.8, 'train_img': './data/sample/train_data/train_images', 'train_txt': './data/sample/train_data
Tensorflow 错误001：CUDA_ERROR_OUT_OF_MEMORY
大可的杨先森的博客
11-14 1431
代码出现tensorflow.python.framework.errors_impl.InternalError，from device: CUDA_ERROR_OUT_OF_MEMORY 此文为解决机器学习中使用tensorflow，在运行代码出现上述报错情况 1.先运行nvidia-smi 检查GPU运行情况，若内存够用进入2 2.代码应作已下修改 import tensorflo...
Tensorflow: CUDA_ERROR_OUT_OF_MEMORY
Felaim的博客
11-08 7808
最近LZ一直在使用tensorflow,遇到有些小问题,就遇到一个,解决一个,有些一直重复出现,或者是做个笔记,加快小伙伴的使用. 在写测试程序代码的时候出现如下报错,源代码就不贴出来了… 2018-11-07 17:10:30.875135: E tensorflow/stream_executor/cuda/cuda_driver.cc:924] failed to allocate 62.2...
Python tensorflow报错： CUDA_ERROR_OUT_OF_MEMORY
qq_34851605的博客
11-20 3636
tensorflow报错： CUDA_ERROR_OUT_OF_MEMORY 这几天在做卷积神经网络的一个项目，遇到了一个问题CUDA_ERROR_OUT_OF_MEMORY。运行代码时前三四百次都运行正常，之后就一直报这个错误（而且第二次、第三次重新运行程序时，报错会提前），但是程序不停止。今天空闲下来，就看一看 这个问题。 详细信息（因为我的报错解决了，所以从网上找了一个相同的报错信息）： 2017-03-28 23:38:19.485396: E tensorflow/stream_executor/
Keras/TensorFlow报错CUDA_ERROR_OUT_OF_MEMORY:
Kyrielong的博客
03-07 6371
用keras的model.predict时报错如下： failed to allocate 4.00G (4294967296 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory 2019-03-07 10:24:00.610883: E tensorflow/stream_executor/cuda/cuda_driver.c...
failed to allocate 18.41M (19300352 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY
qq_36362060的博客
06-13 1万+
tensorflow 训练时出现failed to allocate 18.41M (19300352 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY 我在ubuntu16.04中安装的是tensorflow-gpu1.6.0版本的，在使用jupyter notebook训练时出现了failed to allocate 18.41M (19300352 ...
Encountered unexpected error during CUDA execution. The CUDA error was: CUDA_ERROR_OUT_OF_MEMORY
最新发布
10-24
嗯，用户遇到了CUDA_ERROR_OUT_OF_MEMORY错误，这是GPU显存不足导致的常见问题。用户提供的引用材料主要聚焦在TensorFlow和FFmpeg场景，但实际解决方案具有通用性。 用户可能正在本地调试深度学习模型或视频处理...
TensorFlow2.0 CUDA_ERROR_OUT_OF_MEMORY out of memory 解决办法
胭脂草的ABC博客
08-10 2994
今天头一次使用TensorFlow2.0的GPU版本进行训练，训练中遇到【CUDA_ERROR_OUT_OF_MEMORY out of memory】，于是百度各种解决办法，基本能查到的都是TensorFlow1.0的用法，但此方法在TensorFlow2.0中不适用。于是经过多方查找，终于找到设置方法，在此进行记录，一是方便日后回看，二是弥补百度中各种查不到。 TensorFlow1.0用法 from keras import backend as K config = t...
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码： with torch.no_grad(): # 停止自动反向计算梯度 参考： https://wisdomai.xyz/tool/pytorch/archives/2301 https://ptorch.com/news/160.html 作者：菜叶儿掉啦
【Tensorflow-Error】CUDA_ERROR_OUT_OF_MEMORY: out of memory
热门推荐
luckynote
03-20 1万+
在Tensorflow 训练模型时报错提示： failed to allocate 3.77G (4046333952 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory 虽然会报出显存溢出问题，但不影响正常训练，不过笔者还是想知道这个问题是怎么来的。 废话不多说，先上session初始化的代码 gpu_optio...
Tensorflow: CUDA_ERROR_OUT_OF_MEMORY 亲测有效
种花家的奋斗兔的博客
11-02 2015
import os os.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.5 # maximun alloc gpu50% of MEM config.gpu_optio...
TensorFlow中出现 CUDA_ERROR_OUT_OF_MEMORY
weixin_42608406的博客
03-22 274
参考 https://blog.youkuaiyun.com/qq_34914551/article/details/86604988 我设置后还是没有解决问题，又把batch_size改小了点才解决
TensorFlow中出现 CUDA_ERROR_OUT_OF_MEMORY 的错误解决方案
我的博客有点东西
01-23 5190
我是在用谷歌开源的facenet工程，使用自带的MTCNN用来align人脸数据集时遇到这个问题的。 这个问题出现的原因在于TF想获得全部的GPU资源，设总资源为M，现在只能为TF分配N，N<M，然后TF就不干了。 对于facenet这个工程来说，只需把arg中的可选参数 gpu-memory-fraction 设为0.7，反正必须小于1，是能N/M的百分比。   也可以这么做 ...
深度学习Tensorflow: CUDA_ERROR_OUT_OF_MEMORY解决办法
菜鸡正在学习
09-08 1069
目前在用深度学习训练，训练中设置batch size后可以正常跑通，但是在训练一轮save_model时，总出现这个错误，即使我调batch size到1也依旧会报错。默认调用最大GPU(但我的GPU显存不足以满足需求，所以报错)。经过查询，再logger.py调用tf处，加入。查询后了解到是因为TensorFlow中的。发现是在 调用logger时出现问题。
通过tensorflow调用CUDA库提示“CUDA_ERROR_OUT_OF_MEMORY: out of memory”的问题的解决办法
02-10 2283
设置tensorflow动态申请显存。 具体方法见： https://devtalk.nvidia.com/default/topic/1068031/cudnn/geforce-gtx-1660-super-cuda-not-working-in-anaconda/post/5411179/#5411179 对于tf1，可以进行如下设置： import keras impor...
[BUG | CUDA_ERROR_OUT_OF_MEMORY]
weixin_45714660的博客
04-25 949
解决此问题的一种方法是释放不需要的 GPU 内存，并减少程序中使用的内存量。例如，您可以使用 PyTorch 的 .cuda() 方法将张量移动到 GPU 上，然后使用 .cpu() 方法将其移回 CPU，以释放 GPU 内存。另一种方法是使用更高端的 GPU 或增加 GPU 内存。您可以考虑在可行的情况下升级到更高端的 GPU，或者添加更多的 GPU 内存。这通常是因为您尝试在 GPU 上分配大量内存，但 GPU 内存已经耗尽。您可以通过检查当前系统上可用的 GPU 内存量来验证这一点。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

得食猫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值