tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现
作者:ming.zhang 时间:2021-06-03 17:02:32
在Python代码中指定GPU
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
设置定量的GPU使用量:
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存
session = tf.Session(config=config)
设置最小的GPU使用量:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
来源:https://blog.csdn.net/u010397980/article/details/84257734
标签:tensorflow,gpu,百分比,使用量
0
投稿
猜你喜欢
Python 统计位数为偶数的数字代码详解
2023-12-05 19:43:03
简单了解Python3 bytes和str类型的区别和联系
2023-11-19 21:52:29
python中的各种运算符介绍
2021-08-09 12:49:04
Python中列表与元组的乘法操作示例
2021-05-09 17:11:25
QQ影音感念亲恩皮肤,不只是大按钮这么简单
2009-01-04 14:16:00
Python+Appium新手教程
2022-12-01 04:12:44
详解PHP的引用计数
2023-11-14 18:00:34
Python3.8.2安装包及安装教程图文详解(附安装包)
2021-11-17 19:35:14
Python命令行参数解析模块optparse使用实例
2023-11-04 08:09:08
php验证码的制作思路和实现方法
2023-09-04 13:23:37
解读MySQL的InnoDB引擎日志工作原理
2011-01-04 19:59:00
Python NLP开发之实现聊天机器人
2021-12-21 18:04:48
2个asp获取安全字符串的方法
2007-10-19 15:08:00
一文搞懂Python中is和==的区别
2023-11-15 09:42:27
Mootools 1.2教程(14)——定时器和哈希简介
2008-12-08 12:50:00
Python使用dict.fromkeys()快速生成一个字典示例
2022-05-10 08:13:23
使用python BeautifulSoup库抓取58手机维修信息
2022-08-10 01:55:20
Mysql中limit的用法方法详解与注意事项
2023-11-16 09:20:05
利用OpenCV实现YOLO对象检测方法详解
2021-01-21 06:34:15
利用javaScript实现点击输入框弹出窗体选择信息
2023-09-13 03:14:50