Tensorflow取消占用全部GPU
2024-08-25 20:58:42
参考:https://www.cnblogs.com/jiu0821/p/9501665.html
Tensorflow默认是会占用全部的GPU,而有时候你根本不需要那么占用那么多GPU资源,这时候就可以采用如下三种方法进行解决:
1、在构造tf.Session()时候通过传递tf.GPUOptions作为可选配置参数的一部分来显式地指定需要分配的显存比例,如下所示:
# 假如有12GB的显存并使用其中的4GB:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
per_process_gpu_memory_fraction指定了每个GPU进程中使用显存的上限,但它只能均匀作用于所有GPU,无法对不同GPU设置不同的上限
2、尝试如下设置:
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
当allow_growth设置为True时,分配器将不会指定所有的GPU内存,而是根据需求增长
3、指定GPU编号:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" 或者在脚本或者命令行中指定
export CUDA_VISIBLE_DEVICES=0
最新文章
- JavaScript 函数表达式
- Unity4、Unity5移动平台多线程渲染在部分安卓手机上会造成闪退
- 1、C语言基本数据类型
- SAE上传文件到storage
- 安装php时的配置选项
- Leetcode Sqrt(x)
- word - 如何让 图片任意移动
- select multiple images in Android Gallery
- 用python做些有意思的事——分析QQ聊天记录——私人订制
- CentOS学习笔记—软件管理程序RPM、YUM
- 【BZOJ】【3280】小R的烦恼
- Git服务器搭建全过程分步详解【转】
- IDEA使用的点点滴滴
- QQ群成员提取
- NDK如何调试系统核心动态库(无系统源码的情况)
- vue-axios基本用法
- scrapy数据存储在mysql数据库的两种方式
- Python爬虫(一)——开封市58同城租房信息
- 剑指Offer 27. 字符串的排列 (字符串)
- BEAM188简单应用
热门文章
- Http协议和web本职【转自丁码农】
- SAP Customer Data Cloud(Gigya)的用户搜索实现
- request.getParameterMap和request.getParameter不一样的显示
- 在Vue-cli项目中引入Bootstrap
- BIND简易教程(0):在Ubuntu下源码安装BIND(其实跟前面的教程没太大关系)
- UVa 11582 - Colossal Fibonacci Numbers!(数论)
- gluoncv 目标检测,训练自己的数据集
- Qgis里的查询过滤
- 给已安装的NGINX添加新的模块
- 如何安装zip格式的MySQL