原文

tensorflow ConfigProto

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

控制GPU资源使用率相关推荐

jQuery.NiceScroll - 有史以来最好的 nicescroll 版本——在现代浏览器和移动设备上极其流畅和一致，资源使用率低(中文文档)
jQuery.NiceScroll 特征依赖关系使用配置参数有史以来最好的 nicescroll 版本--在现代浏览器和移动设备上极其流畅和一致,资源使用率低官网:nicescroll.ar ...
10槽PCIE4.0扩展坞可实现GPU资源动态分配
PCIE4.0扩展坞,速度更快,GPU资源可动态分配日本JSA公司推出全新的PCIe4.0 X16扩展坞设备, 采用两个抽屉式结构, 每个抽屉能安装4个双宽的270mm长PCIe卡, 外加机箱上2个 ...
OBS Studio 窗口采集game-capture注入之OpenGL与D3D11的GPU资源进行互操作
OBS Studio 窗口采集game-capture注入之OpenGL与D3D11的GPU资源进行互操作 OBS Studio 窗口采集game-capture注入之OpenGL与D3D11的GPU ...
还怕GPU资源不够用？多实例GPU MIG助攻资源利用难题
从农业时代的锄头镰刀,到工业时代蒸汽机发电机,每个时代都有每个时代的象征符.在迈向工业 4.0 智能化时代的现在,转战 GPU.使能并行计算.扩展云上算力...... 数字世界的每一个关键词无一不是围 ...
浪潮ai服务器最新数据,浪潮发布的最新AI服务器，将GPU资源利用率提升至前所未有的水平...
浪潮在ISC20大会期间发布支持最新NVIDIA® Ampere架构A100 PCIe Gen4的AI服务器NF5468M6和NF5468A5,为AI用户提供兼具超强性能与灵活性的极致AI计算平台. ...
如何在kubernetes中使用共享GPU资源
目录背景 Kubernetes如何使用物理GPU Kubernetes如何使用共享GPU算力背景作为推动人工智能技术进步的"三驾马车",算法.数据和计算力在过去的5-10年间 ...
GPU 资源消耗原因和解决方案：
GPU 资源消耗原因和解决方案: GPU能干的事情比较单一:接收提交的纹理Texture
总奖金 200 万的 AI Challenger 开赛，可申请免费 GPU 资源
记者 | 周翔 8 月 14 日,创新工场.搜狗和今日头条联合宣布共同发起"AI Challenger 全球 AI 挑战赛".其中,CSDN 作为选手社区,为大赛提供支持.(更多赛 ...
Docker 4 Docker安全（cgroup控制容器资源），lxcfs安全加固，docker-machine部署及配置，docker-compose的部署
1.使用内核自带Cgroups 控制docker资源上限: 1.1: pwd :/sys/fs/cgroup/cpu查看docker集成的cgoups控制cpu的命令:(还有memory) [root ...

控制GPU资源使用率

原文

tensorflow ConfigProto

控制GPU资源使用率

控制使用哪块GPU

控制GPU资源使用率相关推荐

最新文章

热门文章