tensorflow ConfigProto
2024-08-26 12:16:20
tf.ConfigProto一般用在创建session的时候。用来对session进行参数配置 with tf.Session(config = tf.ConfigProto(...),...)
#tf.ConfigProto()的参数
log_device_placement=True : 是否打印设备分配日志
allow_soft_placement=True : 如果你指定的设备不存在,允许TF自动分配设备
tf.ConfigProto(log_device_placement=True,allow_soft_placement=True) 控制GPU资源使用率
#allow growth
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
# 使用allow_growth option,刚一开始分配少量的GPU容量,然后按需慢慢的增加,由于不会释放
#内存,所以会导致碎片 # per_process_gpu_memory_fraction
gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
session = tf.Session(config=config, ...)
#设置每个GPU应该拿出多少容量给进程使用,0.4代表 40% 控制使用哪块GPU
~/ CUDA_VISIBLE_DEVICES=0 python your.py#使用GPU0
~/ CUDA_VISIBLE_DEVICES=0,1 python your.py#使用GPU0,1
#注意单词不要打错 #或者在 程序开头
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用 GPU 0
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0,1 ---------------------
作者:ke1th
来源:CSDN
原文:https://blog.csdn.net/u012436149/article/details/53837651
版权声明:本文为博主原创文章,转载请附上博文链接!
最新文章
- 2000条你应知的WPF小姿势 基础篇<;63-68 Triggers和WPF类逻辑结构>;
- [No000070]Flash与C#通信
- BZOJ 1050 旅行comf(枚举最小边-并查集)
- WCF Windows Service Using TopShelf and ServiceModelEx z
- Android ImageView 详述
- MVC5 自定义表单错误信息
- 【转】Hibernate利用@DynamicInsert和@DynamicUpdate生成动态SQL语句
- Codeforces Round #316 div2
- 图片与文字在div里实现垂直水平都居中
- 微信小程序大全(上)(最新整理 建议收藏)
- JDK动态代理[1]----代理模式实现方式的概要介绍
- Baidu音乐爬虫
- Nginx服务器导致CSS无法解析不起效果
- Webpack友好的错误提示插件friendly-errors-webpack-plugin
- B. Menci 的序列
- Mybatis常用代码
- JSON库的使用研究(一)
- java String、StringBuffer、StringBuild、StringTokenizer
- poj3155 最大密度子图
- CSS sprite使用
热门文章
- AutoIT: GUISetFont VS GUICtrlSetFont
- 利用Redis Sentinel实现redis主从自动切换
- hdu4608 I-number
- tools:context=";.MainActivity的作用 (转载)
- Linux系统挂载NTFS文件系统(转载)
- node+ws模块实现websocket
- Ruby&#160;数式匹配器
- 景女神与她的托福(BFS+状压)
- Six degrees of Kevin Bacon
- Spring 中 ApplicationContext 和 BeanFactory 的区别