tensorflow 如何限制显存大小
2024-08-28 09:08:45
Python在用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)
最新文章
- WinSetupFromUSB - 制作多系统U盘安装All-In-One的利器
- appium V1.5.x变化
- WebStorm 2016.2 破解方法
- linux小程序--cmatrix
- POJ Ant Counting DP
- Java基础之写文件——通过缓冲流写文件(StreamOutputToFile)
- [开发笔记]-FireWorks常用操作快捷键
- CSS之text-stroke
- vs2005用正则表达式统计有效代码行数
- ios 设置label的高度随着内容的变化而变化
- C# 邮件发送注意事项
- HDU 4288 Coder (线段树)
- React.js再探(二)
- php-fpm开机启动
- ASP.NET Core 2 High Performance 目录和读书笔记
- java 字符与ASCII码互转
- JS DOM操作(五) Window.docunment对象——操作元素
- 什么是IO多路复用?Nginx的处理机制
- php aes加密
- Mac 10.12高级防火墙pfctl教程收集(待实践)