Tensorflow设置显存自适应,显存比例
来源:互联网 发布:室外装修设计软件 编辑:程序博客网 时间:2024/04/27 22:33
Tensorfow框架下,在模型运行时,设置对显存的占用。
1. 按比例
config = tf.ConfigProto()config.gpu_options.per_process_gpu_memory_fraction = 0.4 # 根据自己的需求确定session = tf.Session(config=config, ...)
2. 自适应
config = tf.ConfigProto()config.gpu_options.allow_growth = Truesession = tf.Session(config=config, ...)
设置GPU的使用率的时候,都是在创建Session的时候,对config类进行设置。
此外,当电脑上有多块GPU的时候,可以指定选取哪一快GPU进行计算。
# 在程序开头添加
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用 GPU 0 0对应着ubuntu系统给GPU的序号,可通过Nvidia-smi命令查看若存在多个GPU
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0,1
一个常见的在代码中指定GPU使用的范例:
import tensorflow as tfimport numpy as npimport osos.environ['CUDA_VISIBLE_DEVICES']='0'config = tf.ConfigProto()config.gpu_options.per_process_gpu_memory_fraction=0.6
with tf.Session(graph=...,config=config) as sess:
## 后续的操作
阅读全文
0 0
- Tensorflow设置显存自适应,显存比例
- tensorflow设置显存自适应和显存比例.
- Tensorflow设置显存自适应,显存比例
- Tensorflow设置显存自适应,显存比例
- TensorFlow显存设置
- TensorFlow指定特定GPU以及占用显存的比例
- tensorflow GPU显存控制
- BIOS显存设置
- 【Ubuntu-Tensorflow】GPU设置及显存资源分配
- tensorflow中指定GPU及GPU显存设置
- tensorflow入门(二)---tensorflow显存管理
- 监视显存
- tensorflow使用GPU训练时的显存占用问题
- tensorflow使用GPU训练时的显存占用问题
- tensorflow中使用指定的GPU及GPU显存
- tensorflow使用GPU训练时的显存占用问题
- tensorflow显存、载入模型、优化器(个人笔记)
- 显卡显存和内存显存的区别
- 8.对象
- 利用Python进行数据分析笔记-第四章Numpy
- 给定一个整数数组,1≤a [i]≤n(n =数组的大小),一些元素出现两次,其他出现一次 查找在该数组中出现两次的所有元素
- eclipse创建webservice-简单教程案例
- Java设计模式百例
- Tensorflow设置显存自适应,显存比例
- Elasticsearch 5.x分布式集群
- centos7下Nginx配置(一)
- SYS_CONNECT_BY_PATH 函数综合应用 应用多字段拼在一行字符串
- Java练习--泛型(12)
- FFM模型在点击率预估中的应用实践
- mybatis-generator自动生成代码工具
- 进程同步实例
- JAVA设计模式之解释器模式