脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → python 限制GPU和CPU占用

运行tensorflow python程序,限制对GPU和CPU的占用操作

 更新时间：2020年02月06日 09:48:35   作者：keep_forward  
今天小编就为大家分享一篇运行tensorflow python程序,限制对GPU和CPU的占用操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

一般情况下，运行tensorflow时，默认会占用可以看见的所有GPU，那么就会导致其它用户或程序无GPU可用，那么就需要限制程序对GPU的占用。并且，一般我们的程序也用不了所有的GPU资源，只是强行霸占着，大部分资源都不会用到，也不会提升运行速度。

使用nvidia-smi可以查看本机的GPU使用情况，如下图，这里可以看出，本机的GPU型号是K80，共有两个K80，四块可用（一个K80包括两块K40）。

1、如果是只需要用某一块或某几块GPU，可以在运行程序时，利用如下命令运行：CUDA_VISIBLE_DEVICES=0,1 python test.py

这里表示只有GPU 0和1对程序可见，因此也就限制了程序只能用GPU 0和1

同样，也可以在代码里指定

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

如果想只用CPU，不用CPU来运行程序，可以用如下命令(所有GPU都不可见)：

CUDA_VISIBLE_DEVICES='' python test.py

或者是

CUDA_VISIBLE_DEVICES="-1" python test.py

2、让tensorflow只按需索取显存，如下代码所示

#only minimum use gpu
gpu_config = tf.ConfigProto()
gpu_config.gpu_options.allow_growth = True
with tf.Session(config = gpu_config) as sess:

前面是对GPU的限制，那如果不用GPU，只用CPU呢？如何限制对CPU的使用呢？

前面也有提到，如果使用命令CUDA_VISIBLE_DEVICES=“”python test.py可以只使用CPU，那如果想只使用部分CPU呢?可以通过如下代码限制

cpu_config = tf.ConfigProto(intra_op_parallelism_threads = 8, inter_op_parallelism_threads = 8, device_count = {'CPU': 8})
with tf.Session(config = cpu_config) as sess:

以上这篇运行tensorflow python程序,限制对GPU和CPU的占用操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python人工智能之混合高斯模型运动目标检测详解分析

    Python人工智能之混合高斯模型运动目标检测详解分析

    运动目标检测是计算机视觉领域中的一个重要内容，其检测效果将会对目标跟踪与识别造成一定的影响，本文将介绍用Python来进行混合高斯模型运动目标检测，感兴趣的朋友快来看看吧
    2021-11-11
  • django 捕获异常和日志系统过程详解

    django 捕获异常和日志系统过程详解

    这篇文章主要介绍了django-捕获异常和日志系统过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-07-07
  • 快速下载VScode并配置Python运行环境(图文教程)

    快速下载VScode并配置Python运行环境(图文教程)

    本文主要介绍了快速下载VScode并配置Python运行环境，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-05-05
  • python制作机器人的实现方法

    python制作机器人的实现方法

    机器人自动回复在很多场景中都可以用的上，本文主要介绍了python制作机器人的实现方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-06-06
  • python矩阵的转置和逆转实例

    python矩阵的转置和逆转实例

    今天小编就为大家分享一篇python矩阵的转置和逆转实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12
  • Opencv实现二维直方图的计算及绘制

    Opencv实现二维直方图的计算及绘制

    这篇博客将介绍如何使用Opencv进行二维直方图的计算及绘制，维直方图可以让我们对不同的像素密度有更好的了解，感兴趣的可以了解一下
    2021-07-07
  • Python多线程与同步机制浅析

    Python多线程与同步机制浅析

    线程（Thread）是操作系统能够进行运算调度的最小单位；线程自己不拥有系统资源，只拥有一点儿在运行中必不可少的资源，但它可与同属一个进程的其它线程共享进程所拥有的全部资源
    2022-12-12
  • Python自动化测试Eclipse+Pydev 搭建开发环境

    Python自动化测试Eclipse+Pydev 搭建开发环境

    本文主要介绍python自动化测试环境搭建,这里对Eclipse+Pydev 搭建开发环境做了图文详解，有需要的小伙伴可以参考下
    2016-08-08
  • Python实现监控内存使用情况和代码执行时间

    Python实现监控内存使用情况和代码执行时间

    我的代码的哪些部分运行时间最长、内存最多？我怎样才能找到需要改进的地方？在开发过程中，我很确定我们大多数人都会想知道这一点。本文总结了一些方法来监控 Python 代码的时间和内存使用情况，希望对大家有所帮助
    2023-01-01
  • 通过Python OpenGL的point sprite技术绘制雪花

    通过Python OpenGL的point sprite技术绘制雪花

    通常，点精灵（point sprite）技术被用于描述大量粒子在屏幕上的运动，自然也可以用于绘制雪花。本文将通过Python OpenGL绘制雪花，感兴趣的可以动手试一试
    2022-02-02

最新评论