tensorflow内存问题

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/atzhangzt/article/details/82490592

tensorflow在训练时默认占用所有GPU的显存,提示内存已满!


1. 动态分配内存
当allow_growth设置为True时,分配器将不会指定所有的GPU内存,而是根据需求增长

config = tf.ConfigProto()  
config.gpu_options.allow_growth=True  
sess = tf.Session(config=config)

2. 设置GPU内存使用率
显式地指定需要分配的显存比例,假如有12GB的显存并使用其中的4GB:如下所示:


gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

3、在执行训练脚本前使用:
来限制可见的GPU数目

export CUDA_VISIBLE_DEVICES=1
展开阅读全文

内存问题

10-19

[code=C/C++]rn//编译和连接能通过,但运行时候提示内存错误!大家帮忙看看吧!rn//哪有问题呢?大家提提意见吧!rn#include rn#include rnrnusing namespace std;rnrntypedef struct nodernrn int data;rn struct node *next;rnQnode;rnrnrnclass Queuernrnpublic:rn Queue();rn void InitQueue();rn void show()const;rn void EnQueue(int e);rn void DeQueue(int &e);rn ~Queue();rnprivate:rn Qnode *rear;rn Qnode *front;rn;rnrnQueue::Queue()rnrn front=new node;rn rear=front;rn front->next=NULL;rnrnrnvoid Queue::InitQueue()rnrn Qnode *p2=new node;rn int i=0;rn cout<<"请输入数值(-1 退出): ";rn cin>>p2->data;rn while (p2->data!=-1)rn rn ++i;rn if(i==1)rn rn front->next=p2; //队首指针指向队首rn rear->next=p2;rn rear=p2;rn rn elsern rn rear->next=p2;rn rear=p2;rn rn p2=new node;rn cin>>p2->data;rn rn rear->next=NULL;rnrnrnrnvoid Queue::show() constrnrnrn if(front==rear)rn cout<<"此队为空队!"<next;rn cout<<"顺序的队值为: "<data<<" ";rn p=p->next;rn rn cout<data=e;rn p->next-=NULL;rn rear->next=p;rn rear=p;rnrnrnvoid Queue::DeQueue(int &e) //从队首删除一个元素rnrn if (front==rear)rn cout<<"此队为空队!"<next;rn e=p->data;rn front->next=p->next;rn if(rear==p) rn rear=front;rn delete p;rn rnrnrn Queue::~Queue()rnrn while(front)rn rn rear=front->next;rn delete front;rn front=rear;rn rnrnrnint main ()rnrn Queue Q;rn int e;rn Q.InitQueue();rn cout<<"请输入要进队数值:"<>e;rn Q.EnQueue(e);rn Q.DeQueue(e);rn cout<<"出队数值: "< 论坛

没有更多推荐了,返回首页