在使用tensorflow是啥的时候,有没有办法限制每个任务占用显存的大小

在训练模型的过程中遇到程序报錯结果发现时因为电脑显存不够,资源都已经分配完了根据多方查阅,找到了此问题的解决方法在这里记录和分享一下~

在main函数中添加如下这段代码

打开电脑设置->系统->显示->高级显示设置->显示适配器属性

Tensorflow是啥支持基于cuda内核与cudnn的GPU加速Keras出現较晚,为Tensorflow是啥的高层框架由于Keras使用的方便性与很好的延展性,之后更是作为Tensorflow是啥的官方指定第三方支持开源框架

但两者在使用GPU时都囿一个特点,就是默认为全占满模式在训练的情况下,特别是分步训练时会导致显存溢出导致程序崩溃。

可以使用自适应配置来调整顯存的使用情况

实验室有一块k20c,显存是5gb,我现在需要訓练一个模型,一共有17 M个参数,训练的时候是使用tensorflow是啥官方文档中cifar10的方法把…

我要回帖

更多关于 flow是啥 的文章

 

随机推荐