脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → 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 pytest自动化测试库十个强大用法示例

    Python pytest自动化测试库十个强大用法示例

    本文将介绍Python的pytest库的10个强大用法,并提供相应的代码示例,帮助你更好地理解和应用单元测试,它提供了许多高级功能和便利的用法,能够让我们更轻松地编写和执行单元测试
    2024-01-01
  • Python内置函数—vars的具体使用方法

    Python内置函数—vars的具体使用方法

    本篇文章主要介绍了Python内置函数—vars的具体使用方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2017-12-12
  • pycharm实现在虚拟环境中引入别人的项目

    pycharm实现在虚拟环境中引入别人的项目

    这篇文章主要介绍了pycharm实现在虚拟环境中引入别人的项目，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-03-03
  • 使用Keras加载含有自定义层或函数的模型操作

    使用Keras加载含有自定义层或函数的模型操作

    这篇文章主要介绍了使用Keras加载含有自定义层或函数的模型操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • Python爬取读者并制作成PDF

    Python爬取读者并制作成PDF

    本文是在学习了beautifulsoup之后，制作的一个爬取读者杂志并使用reportlab制作成pdf的python小工具，咱也文艺一下：），分享给大家，有需要的小伙伴参考下吧。
    2015-03-03
  • Python实现将Excel转换成xml的方法示例

    Python实现将Excel转换成xml的方法示例

    这篇文章主要介绍了Python实现将Excel转换成xml的方法,涉及Python针对Excel文件的读取、遍历、节点设置与xml生成等相关操作技巧,需要的朋友可以参考下
    2018-08-08
  • python+opencv3.4.0 实现HOG+SVM行人检测的示例代码

    python+opencv3.4.0 实现HOG+SVM行人检测的示例代码

    这篇文章主要介绍了python+opencv3.4.0 实现HOG+SVM行人检测的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-01-01
  • Python爬虫之urllib库详解

    Python爬虫之urllib库详解

    大家好，本篇文章主要讲的是Python爬虫之urllib库详解，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下
    2022-02-02
  • Python中网络请求中Retry策略实现方式

    Python中网络请求中Retry策略实现方式

    这篇文章主要介绍了Python中网络请求中Retry策略实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-06-06
  • PyQt5+QtChart实现柱状图的绘制

    PyQt5+QtChart实现柱状图的绘制

    QChart是一个QGraphicScene中可以显示的QGraphicsWidget。本文将利用QtChart实现柱状图的绘制，文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下
    2022-12-12

最新评论