TensorFlow 2.0之后动态分配显存方式
更新时间：2022年12月08日 09:28:04 作者：scluis
这篇文章主要介绍了TensorFlow 2.0之后动态分配显存方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
TensorFlow 2.0之后动态分配显存
import tensorflow as tf config = tf.compat.v1.ConfigProto() config.gpu_options.allow_growth = True sess = tf.compat.v1.Session(config = config) tf.compat.v1.keras.backend.set_session(sess)
keras和tensorflow动态分配显存
tensoflow或者基于tensorflow的keras，好像默认会直接将GPU中所有的显存都读过来，然后慢慢用。
这样如果是服务器是的话，别人就没法用了，这里转载了动态分配显存的方法：
tensorflow下，加入代码：
config = tf.ConfigProto() config.gpu_options.allow_growth = True #设置动态分配显存 session = tf.Session(config=config, ...)
或者：
config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.3 #只一次调用30%的显存 session = tf.Session(config=config, ...)
对于keras，如果使用tensorflow作为后端，多加一句话就好了：
import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" from keras.backend.tensorflow_backend import set_session config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.3 set_session(tf.Session(config=config)) # 此处不同
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了在Python中使用列表生成式的教程,列表生成式是Python具有的重要特性,需要的朋友可以参考下2015-04-04
大家好，本篇文章主要讲的是python pytorch图像识别基础介绍，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-02-02
range函数大多数时常出现在for循环中,在for循环中可做为索引使用,下面这篇文章主要给大家介绍了关于Python中range函数的基本用法,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-01-01
实际开发中我们可能会遇到一类问题，如何将Pandas DataFrame转换为字典，本文就来介绍一下，感兴趣的可以了解一下2021-05-05
这篇文章主要为大家想想介绍了Python中Flask数据库的连接与使用，文中的示例代码讲解详细，具有一定的借鉴价值，感兴趣的小伙伴可以学习一下2023-02-02
这篇文章主要介绍了Python基础类继承重写实现原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-04-04
python导出chrome书签到markdown文件的实例代码
python导出chrome书签到markdown文件，主要就是解析chrome的bookmarks文件，然后拼接成markdown格式的字符串，最后输出到文件即可。下面给大家分享实例代码，需要的朋友参考下2017-12-12
这篇文章主要介绍了Python如何优雅删除字符列表空字符及None元素,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
在python的输出结果中，尤其是浮点数的输出，当我们需要写入文本文件时，最好是采用统一的输出格式，这样也能够增强结果的可读性，这篇文章主要介绍了控制Python浮点数输出位数的方法,需要的朋友可以参考下2022-04-04
使用Python和scikit-learn创建混淆矩阵的示例详解
这篇文章主要介绍了使用Python和scikit-learn创建混淆矩阵的示例详解,该示例包括生成数据集、为数据集选择合适的机器学习模型、构建、配置和训练它，最后解释结果，即混淆矩阵，需要的朋友可以参考下2022-06-06
最新评论