脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Tensorflow显存自适应,显存比例

Tensorflow设置显存自适应,显存比例的操作

 更新时间：2020年02月03日 09:58:38   作者：tsq292978891  
今天小编就为大家分享一篇Tensorflow设置显存自适应,显存比例的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

Tensorfow框架下，在模型运行时，设置对显存的占用。

1. 按比例

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4 # 根据自己的需求确定
session = tf.Session(config=config, ...)

2. 自适应

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

设置GPU的使用率的时候，都是在创建Session的时候，对config类进行设置。

此外，当电脑上有多块GPU的时候，可以指定选取哪一快GPU进行计算。

# 在程序开头添加
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用 GPU 0 0对应着ubuntu系统给GPU的序号,可通过Nvidia-smi命令查看

若存在多个GPU

os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0，1

一个常见的在代码中指定GPU使用的范例：

import tensorflow as tf
import numpy as np
import os

os.environ['CUDA_VISIBLE_DEVICES']='0'
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction=0.6

with tf.Session(graph=...,config=config) as sess:

## 后续的操作

以上这篇Tensorflow设置显存自适应,显存比例的操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • python删除本地夹里重复文件的方法

    python删除本地夹里重复文件的方法

    这篇文章主要为大家详细介绍了python删除本地夹里重复文件的方法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2018-05-05
  • python机器学习deepchecks库训练检查模型特点探索

    python机器学习deepchecks库训练检查模型特点探索

    这篇文章主要介绍了python机器学习deepchecks库的训练检查模型特点实例探索,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2024-01-01
  • Flask中sqlalchemy模块的实例用法

    Flask中sqlalchemy模块的实例用法

    在本篇文章里小编给大家整理了关于Flask中sqlalchemy模块的实例用法，需要的朋友们可以学习下。
    2020-08-08
  • tensorflow建立一个简单的神经网络的方法

    tensorflow建立一个简单的神经网络的方法

    本篇文章主要介绍了tensorflow建立一个简单的神经网络的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2018-02-02
  • 浅析Python四种数据类型

    浅析Python四种数据类型

    在这篇文章里，我们给大家分享了关于Python的四种数据类型相关知识点内容，有兴趣的朋友们参考下。
    2018-09-09
  • 基于PyTorch的permute和reshape/view的区别介绍

    基于PyTorch的permute和reshape/view的区别介绍

    这篇文章主要介绍了基于PyTorch的permute和reshape/view的区别介绍，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • Python PIL读取的图像发生自动旋转的实现方法

    Python PIL读取的图像发生自动旋转的实现方法

    这篇文章主要介绍了Python PIL读取的图像发生自动旋转的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2019-07-07
  • python实现JAVA源代码从ANSI到UTF-8的批量转换方法

    python实现JAVA源代码从ANSI到UTF-8的批量转换方法

    这篇文章主要介绍了python实现JAVA源代码从ANSI到UTF-8的批量转换方法,涉及Python针对文件操作与编码转换的相关技巧,需要的朋友可以参考下
    2015-08-08
  • 利用Python操作MongoDB数据库的详细指南

    利用Python操作MongoDB数据库的详细指南

    MongoDB是由C++语言编写的非关系型数据库,是一个基于分布式文件存储的开源数据库系统,其内容存储形式类似JSON对象,下面这篇文章主要给大家介绍了关于利用Python操作MongoDB数据库的相关资料,需要的朋友可以参考下
    2023-02-02
  • PyTorch和Keras计算模型参数的例子

    PyTorch和Keras计算模型参数的例子

    今天小编就为大家分享一篇PyTorch和Keras计算模型参数的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-01-01

最新评论