脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Tensorflow训练模型默认占满所有GPU

Tensorflow训练模型默认占满所有GPU的解决方案

 更新时间：2023年05月10日 09:00:56   作者：石头儿啊  
这篇文章主要介绍了Tensorflow训练模型默认占满所有GPU的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

Tensorflow训练模型默认占满所有GPU问题

在使用gpu服务器训练tensorflow模型时，总是占满显存！

TensorFlow默认的是占用所有GPU

因此我们需要手动设置使用的GPU编号以及单个GPU显存占用比例

1.第一步需要在代码中开头加入 

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 按照PCI_BUS_ID顺序从0开始排列GPU设备
os.environ["CUDA_VISIBLE_DEVICES"]=‘0'  # 使用0号gpu（想使用其他编号GPU，对应修改引号中的内容即可）
os.environ["CUDA_VISIBLE_DEVICES"]=‘0,1' # 使用0号GPU和1号GPU

2.第二步需要将代码中的sess = tf.Session()改为

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333) # 通过改变0.333可以改变占用显存比例
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

per_process_gpu_memory_fraction=0.333代表的含义就是每个GPU进程中使用显存的上限为该GPU总量的1/3

3.如果想要在程序运行过程中连续查看GPU信息

可以在终端使用该 指令（执行指令：watch -n 3 -d nvidia-smi # 每隔三秒输出一次）（前提是设备中有合适的NVIDIA驱动）

解决tensorflow2.2把GPU显存占满

安装了tensorflow-gpu后，运行程序默认是把GPU的内存全部占满的，有时我们不想全部占满，可以这样操作。

解决代码

import tensorflow as tf 
import os
os.environ['CUDA_VISIBLE_DEVICES']="0" # 指定哪块GPU训练
config=tf.compat.v1.ConfigProto() 
# 设置最大占有GPU不超过显存的80%（可选）
# config.gpu_options.per_process_gpu_memory_fraction=0.8
config.gpu_options.allow_growth = True  # 设置动态分配GPU内存
sess=tf.compat.v1.Session(config=config)

如图：

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python控制台输出俄罗斯方块移动和旋转功能

    Python控制台输出俄罗斯方块移动和旋转功能

    这篇文章主要介绍了Python控制台输出俄罗斯方块移动和旋转功能,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2021-04-04
  • 使用Python控制摄像头拍照并发邮件

    使用Python控制摄像头拍照并发邮件

    这篇文章主要介绍了使用Python控制摄像头拍照并发邮件的相关知识，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-04-04
  • Python使用unittest进行有效测试的示例详解

    Python使用unittest进行有效测试的示例详解

    这篇文章主要介绍了如何使用 unittest 来编写和运行单元测试，希望通过阅读本文，大家能了解 unittest 的基本使用方法，以及如何使用 unittest 中的断言方法和测试用例组织结构
    2023-06-06
  • 使用Python手工计算x的算数平方根,来自中国古人的数学智慧

    使用Python手工计算x的算数平方根,来自中国古人的数学智慧

    本篇采用的计算方法既非二分法也非牛顿迭代法，而是把中国古代的手工计算平方根的方法转成代码来完成。代码有点烦杂，算是抛砖引玉吧，期待高手们写出更好的代码来
    2021-09-09
  • Django实现上传图片功能

    Django实现上传图片功能

    这篇文章为大家详细主要介绍了Django实现上传图片，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-08-08
  • Django时区详解

    Django时区详解

    这篇文章主要介绍了Django时区详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2019-07-07
  • 基于Tensorflow读取MNIST数据集时网络超时的解决方式

    基于Tensorflow读取MNIST数据集时网络超时的解决方式

    这篇文章主要介绍了基于Tensorflow读取MNIST数据集时网络超时的解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • django使用django-apscheduler 实现定时任务的例子

    django使用django-apscheduler 实现定时任务的例子

    今天小编就为大家分享一篇django使用django-apscheduler 实现定时任务的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • Python交换字典键值对的四种方法实例

    Python交换字典键值对的四种方法实例

    字典中有成对出现的键和值,但是字典中的键值对不是都能修改的,只有值才能修改,下面这篇文章主要给大家介绍了关于Python交换字典键值对的四种方法,需要的朋友可以参考下
    2022-12-12
  • python数字图像处理图像的绘制详解

    python数字图像处理图像的绘制详解

    这篇文章主要为大家介绍了python数字图像处理图像的绘制示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-06-06

最新评论