报错解决：ResourceExhaustedError: OOM when allocating tensor with shape

早上在使用tensorflow时遇到如下报错：

Traceback (most recent call last):File "C:\Users\peter\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 786, in runfileexecfile(filename, namespace)File "C:\Users\peter\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfileexec(compile(f.read(), filename, 'exec'), namespace)callbacks = [early_stopping,cp_callback])  # pass callback to trainingFile "C:\Users\peter\Anaconda3\lib\site-packages\tensorflow\python\keras\engine\training.py", line 880, in fitvalidation_steps=validation_steps)File "C:\Users\peter\Anaconda3\lib\site-packages\tensorflow\python\keras\engine\training_arrays.py", line 266, in model_iterationbatch_outs = f(actual_inputs)File "C:\Users\peter\Anaconda3\lib\site-packages\tensorflow\python\keras\backend.py", line 3076, in __call__run_metadata=self.run_metadata)File "C:\Users\peter\Anaconda3\lib\site-packages\tensorflow\python\client\session.py", line 1439, in __call__run_metadata_ptr)File "C:\Users\peter\Anaconda3\lib\site-packages\tensorflow\python\framework\errors_impl.py", line 528, in __exit__c_api.TF_GetCode(self.status.status))ResourceExhaustedError: OOM when allocating tensor with shape[180621,64,1,1981] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc[[{{node conv1d/conv1d/Conv2D}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

这个报错的意思是你的数据太多了，不能一次放进GPU中，虽然你设置了batch_size，但tensorflow默认是一次把所有数据都放进GPU中。
我的解决方案是采用CPU训练

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

报错解决：ResourceExhaustedError: OOM when allocating tensor with shape相关推荐

报错：ResourceExhaustedError OOM when allocating
日萌社人工智能AI:Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战(不定时更新) 报错:ResourceExhaustedError OOM w ...
OP_REQUIRES failed at conv_ops.cc:386 : Resource exhausted: OOM when allocating tensor with shape..
tensorflow-gpu验证准确率是报错如上: 解决办法: 1. 加入os.environ['CUDA_VISIBLE_DEVICES']='2' 强制使用CPU验证-----慢 2.'batch ...
Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos
在CPU下跑的时候并没有报错,换成GPU后就一直OOM,后来被提醒道,显存比内存小很多啊,所以改小了batch_size,并将网络结构进行了优化,减少了词向量的维度和隐藏层节点个数. 另外,在代码里添 ...
报错解决——RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton
问题描述 RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dim ...
Tensorflow CIFAR-10训练例子报错解决
Tensorflow CIFAR-10训练例子报错解决大部分是Api版本问题: AttributeError: 'module' object has no attribute 'SummaryWr ...
zabbix服务器没有.pid文件,PID file /run/zabbix/zabbix_server.pid not readable (yet?) after start. 报错解决...
报错如下: [root@localhost zabbix]# systemctl start zabbix-server Job for zabbix-server.service failed be ...
连接远程服务器CredSSP加密Oracle修正报错解决办法
连接远程服务器CredSSP加密Oracle修正报错解决办法: 打开注册表,快捷输入 "regedit"(类似找命令提示符输入 cmd 一样)找文件夹路径:HKLM(缩写)\S ...
从零开始在ubuntu上安装和使用k8s集群及报错解决
文章目录安装docker 安装kubernetes 配置k8s集群配置虚拟机网络配置Master节点的k8s网络拉取k8s需要的镜像启动 kubeadm 和 kubelet 配置 node节 ...
Linux/ubuntu 安装 redis 4.0报错解决：redis-server.service: Can't open PID file /var/run/redis/redis-server.
此文首发于我的个人博客:Linux/ubuntu 安装 redis 4.0报错解决:redis-server.service: Can't open PID file /var/run/redis/r ...

报错解决：ResourceExhaustedError: OOM when allocating tensor with shape

报错解决：ResourceExhaustedError: OOM when allocating tensor with shape

报错解决：ResourceExhaustedError: OOM when allocating tensor with shape相关推荐

最新文章

热门文章