tensorflow指定GPU与动态分配GPU memory设置


Posted in Python onFebruary 03, 2020

在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。

以上这篇tensorflow指定GPU与动态分配GPU memory设置就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python抓取网页内容示例分享
Feb 24 Python
Python3基础之输入和输出实例分析
Aug 18 Python
Python中下划线的使用方法
Mar 27 Python
Python实现提取谷歌音乐搜索结果的方法
Jul 10 Python
Python 列表排序方法reverse、sort、sorted详解
Jan 22 Python
浅谈python中对于json写入txt文件的编码问题
Jun 07 Python
Python实现iOS自动化打包详解步骤
Oct 03 Python
Python I/O与进程的详细讲解
Mar 08 Python
django使用xadmin的全局配置详解
Nov 15 Python
Django choices下拉列表绑定实例
Mar 13 Python
Python自省及反射原理实例详解
Jul 06 Python
Pandas数据结构之Series的使用
Mar 31 Python
TensorFlow内存管理bfc算法实例
Feb 03 #Python
python numpy数组中的复制知识解析
Feb 03 #Python
opencv python Canny边缘提取实现过程解析
Feb 03 #Python
Pycharm debug调试时带参数过程解析
Feb 03 #Python
Python使用enumerate获取迭代元素下标
Feb 03 #Python
TensorFlow 显存使用机制详解
Feb 03 #Python
opencv python如何实现图像二值化
Feb 03 #Python
You might like
PHP中用header图片地址 简单隐藏图片源地址
2008/04/09 PHP
PHP 循环列出目录内容的函数代码
2010/05/26 PHP
PHP实现连接设备、通讯和发送命令的方法
2015/10/13 PHP
基于php实现的php代码加密解密类完整实例
2016/10/12 PHP
php使用fputcsv实现大数据的导出操作详解
2020/02/27 PHP
对联广告js flash激活
2006/10/19 Javascript
JavaScript中“基本类型”之争小结
2013/01/03 Javascript
JavaScript中this关键词的使用技巧、工作原理以及注意事项
2014/05/20 Javascript
Visual Studio中js调试的方法图解
2014/06/30 Javascript
七夕情人节丘比特射箭小游戏
2015/08/20 Javascript
js倒计时简单实现方法
2015/12/17 Javascript
值得分享的JavaScript实现图片轮播组件
2016/11/21 Javascript
Node.js如何使用Diffie-Hellman密钥交换算法详解
2017/09/05 Javascript
BootStrap Table实现server分页序号连续显示功能(当前页从上一页的结束序号开始)
2017/09/12 Javascript
JavaScript常用工具方法封装
2019/02/12 Javascript
a标签调用js的方法总结
2019/09/05 Javascript
javascript删除数组元素的七个方法示例
2019/09/09 Javascript
基于canvas实现手写签名(vue)
2020/05/21 Javascript
单链表反转python实现代码示例
2018/02/08 Python
python文件操作之批量修改文件后缀名的方法
2018/08/10 Python
Python threading的使用方法解析
2019/08/28 Python
Python获取一个用户名的组ID过程解析
2019/09/03 Python
python 使用递归实现打印一个数字的每一位示例
2020/02/27 Python
python sklearn包——混淆矩阵、分类报告等自动生成方式
2020/02/28 Python
pycharm sciview的图片另存为操作
2020/06/01 Python
Python使用urlretrieve实现直接远程下载图片的示例代码
2020/08/17 Python
HTML5实现Notification API桌面通知功能
2016/03/02 HTML / CSS
Forever 21美国官网:美国标志性快时尚品牌
2017/02/20 全球购物
Dogeared官网:在美国手工制作的珠宝
2019/08/24 全球购物
如何写一个自定义标签
2012/12/28 面试题
网页设计个人找工作求职信
2013/11/28 职场文书
英文推荐信格式范文
2014/05/09 职场文书
表彰大会策划方案
2014/05/13 职场文书
党员组织生活会发言材料
2014/10/17 职场文书
环卫处个人工作总结
2015/03/04 职场文书
python 如何执行控制台命令与操作剪切板
2021/05/20 Python