TensorFlow,Keras限制GPU显存
2024-08-22 15:58:52
运行TensorFlow程序会占用过多的显卡比例,多人共同使用GPU的时候,会造成后面的人无法运行程序。
一、TensorFlow
1.预加载比例限制
tf_config = tensorflow.ConfigProto()
tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50%
session = tensorflow.Session(config=tf_config)
2.自适应
tf_config = tensorflow.ConfigProto()
tf_config.gpu_options.allow_growth = True
session = tensorflow.Session(config=tf_config)
二、Keras
1.当使用Keras的情况下,当import keras时,框架会自动开启一个默认参数的Session。可以通过下面的代码主动创建一个使用了ConfigProto的Session,再注入到Keras的后端中去:
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
set_session(tf.Session(config=config))
注意:第一行必须直接import这个函数,而非keras这个包,而且这句话必须在其他import keras之前,否则keras初始化之后,再替换一个session,原session也不会释放。
三、在Python程序中指定cpu
import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""
最新文章
- java 后台开发关键词解释
- Asp.net 面向接口可扩展框架之消息队列组件
- errno.h
- LEETCODE —— Unique Binary Search Trees [动态规划]
- [Java面试一]面试复习大纲.
- Struts2 源码分析——核心机制
- 找到n中最小的k个数
- objective-c 下面int 和 NSData数据 互相转换的方法
- 安装配置Apache2.4和php7.0
- document.createElement(";A";);
- Python爬虫第一步
- C#7.0
- dp资源分配问题
- ELK 6安装配置 nginx日志收集 kabana汉化
- DataReport使用手记
- 强大的数据库工具 dbForge Studio ForMySql
- SSH免密远程登陆及详解
- mongodb 语句和SQL语句对应(SQL to Aggregation Mapping Chart)
- 第一个django项目-通过命令行和pycharm两种方式
- U盘装win7系统