tensorflow指定GPU与动态分配GPU memory设置
更新时间：2020年02月03日 11:40:18 作者：sherry颖
今天小编就为大家分享一篇tensorflow指定GPU与动态分配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。
以上这篇tensorflow指定GPU与动态分配GPU memory设置就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
本文通过讲解使用基础Python代码来读写和处理CSV文件，点进来的小伙伴一起来学习吧2021-08-08
在本篇文章里小编给大家整理了关于pycharm中显示CSS提示的知识点内容，需要的朋友们可以参考学习下。2019-07-07
又是一年元宵节！如果昨天情人节的红包发得手软又心疼，不妨利用Python OpenGL做一盏花灯送给女朋友，也许比红包更能讨她欢心呢2022-02-02
教你利用python的matplotlib(pyplot)绘制折线图和柱状图
Python绘图需要下载安装matplotlib模块,它是一个数学绘图库,我们将使用它来制作简单的图表,如折线图和散点图,下面这篇文章主要给大家介绍了关于利用python的matplotlib(pyplot)绘制折线图和柱状图的相关资料,需要的朋友可以参考下2022-05-05
本案例通过使用pandas库对电子商务客户数据进行数据预处理,包括数据导入、查看、缺失值处理等处理,具有一定的参考价值,感兴趣的可以了解一下2025-01-01
Python3.7 基于 pycryptodome 的AES加密解密、RSA加密解密、加签验签
这篇文章主要介绍了Python3.7 基于 pycryptodome 的AES加密解密、RSA加密解密、加签验签，本文通过实例代码给大家介绍的非常详细,需要的朋友可以参考下2019-12-12
这篇文章主要介绍了Python堆排序原理与实现方法,结合实例形式详细分析了Python堆排序的概念、原理、实现方法及相关操作注意事项,需要的朋友可以参考下2018-05-05
数字水印技术可以将隐藏信息嵌入到音频文件中而不明显影响音频质量,下面小编将介绍几种在Python中实现音频数字水印的方法,希望对大家有所帮助2025-04-04
这篇文章主要介绍了Python装饰器的应用场景,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-04-04
functools模块是Python标准库中的一个宝库,提供了一些有用的功能,可以帮助您更好地利用函数的潜力,下面小编就来为大家介绍一下functools模块的相关具体使用吧2023-11-11
最新评论