tensorflow指定GPU与动态分配GPU memory设置
作者：sherry颖 发布时间：2023-01-16 11:30:24
标签：tensorflow,GPU,分配,memory
在tensorflow中，默认指定占用所有的GPU，如需指定占用的GPU，可以在命令行中：
export CUDA_VISIBLE_DEVICES=1
这样便是只占用1号GPU，通过命令
nvidia-smi
可以查看各个GPU的使用情况。
另外，也可以在python程序中指定GPU，并且动态分配memory，代码如下
import os
import sys
os.environ['CUDA_VISIBLE_DEVICES'] = sys.argv[1]
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
set_session(tf.Session(config=config))
这样，占用的GPU就是sys.argv[1]了，并且会按需分配memory。
来源：https://blog.csdn.net/sherry_jy/article/details/79564930
0
投稿猜你喜欢
<html> <head> <script type="text/javascript"&- 最近终于找到一个好的方法，使用Python的OpenCV模块识别滑动验证码的缺口，可以将滑动验证码中的缺口识别出来了。 测试使用如
- SQL Server Sa用户相信大家都有一定的理解，下面就为您介绍SQL Server 2000身份验证模式的修改方法及SQL Serve
- 这篇文章主要介绍了python使用配置文件过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可
- 前几天，因为需要实现海外服务端定时停机，涉及到时区的概念。网上搜索了一下，大部分都是谈time.Format中的Layout，非常不成体系，
- 前言近日在做一个报表功能里面有一个这样的需求是统计各部门在某一月入职和离职的人数我的步骤先查出入职的人数SELECT dept ,COUNT
- 在VBScript中有Filter这个函数可以用来对数组进行过滤，并返回原数组的一个子集数组。语法说明： 引用内容Filter 函
- 通过对 26 个字母的设定，设置自己要输出的字体。name = "RUNOOB"# 接收用户输入# name = inp
- 标准的SQL模式匹配SQL的模式匹配允许你使用“_”匹配任何单个字符，而“％”匹配任意数目字符(包括零个字符)。在 MySQL中，SQL的模
- Python TutorPython Tutor 是由 Philip Guo 开发的一个免费教育工具，可帮助学生攻克编程学习中的基础障碍，理
- 通过python与ffmpeg结合使用，可生成进行视频点播、直播的压力测试脚本。可支持不同类型的视频流，比如rtmp或者hls形式。 通过如
- alt的准确含义是，当照片不存在或者load错误时的提示。但同时img也同时支持alt和title，再有某些浏览器的错误解析，因此经常被误导
- 目录前言项目设计后端前端运行项目Q&A前言在前面的Api开发中，我们使用FastApi已经可以很好的实现。但是实际使用中，我们通常建
- 如果是导入有中文的数据，我的mysql 设置的utf8 字符集，所以你要导入的xxx.txt 文件也要保存ut
- 前言Iframe 是一种将网页嵌入到另一个页面的内容中的方法。这是通过使用 HTML 元素、外部网站的 URL 以及窗口在您的网站上的外观参
- 节点类型主要有三种：元素节点，属性节点和文本节点。而对DOM的主要也就是围绕元素节点和属性节点的增删改查。下面就分别从对元素节点的操作和对属
- 错误图片解决mysql -u root -p 以root身份进入mysqlgrant all on *.* to 'root'
- 自定义函数参数传递为 字符串格式 ，传递方式1：用this传递 2：引号缺省 3：转义字符（html中 " 代表"
- 前言本文通过一个小范例来学习java中通过正则表达式如何获得一个字符串中的数字，下面话不多说，来看看详细的介绍吧。示例代码如下：import
- 一直想了解Web编程的技术。PHP是进行Web编程重要的一种语言，书上总是说，PHP是用于服务器端的编程语言。但是，实在不能理解它是怎么用于
