脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → 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 实现矩阵按对角线打印，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-11-11
  • Python中关于列表的常规操作范例以及介绍

    Python中关于列表的常规操作范例以及介绍

    列表是一种有序的集合，可以随时添加和删除其中的元素。在python中使用的频率非常高，本篇文章对大家的学习或工作具有一定的价值，需要的朋友可以参考下
    2021-09-09
  • 在Pycharm中安装Pandas库方法(简单易懂)

    在Pycharm中安装Pandas库方法(简单易懂)

    这篇文章主要介绍了在Pycharm中安装Pandas库方法,文中通过图文介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-02-02
  • kNN算法python实现和简单数字识别的方法

    kNN算法python实现和简单数字识别的方法

    这篇文章主要介绍了kNN算法python实现和简单数字识别的方法,详细讲述了kNN算法的优缺点及原理,并给出了应用实例,需要的朋友可以参考下
    2014-11-11
  • 浅析Python编写函数装饰器

    浅析Python编写函数装饰器

    这篇文章主要介绍了Python编写函数装饰器的相关资料,需要的朋友可以参考下
    2016-03-03
  • 教你如何在pycharm中使用less

    教你如何在pycharm中使用less

    这篇文章主要介绍了如何在pycharm中使用less,操作步骤真的很简单，本文通过图文并茂的形式给大家详细介绍，需要的朋友可以参考下
    2021-10-10
  • Python中字典(dict)和列表(list)的排序方法实例

    Python中字典(dict)和列表(list)的排序方法实例

    这篇文章主要介绍了Python中字典(dict)和列表(list)的排序方法实例,总结来说优先使用内置的sort()方法进行排序,需要的朋友可以参考下
    2014-06-06
  • pytorch dataloader 取batch_size时候出现bug的解决方式

    pytorch dataloader 取batch_size时候出现bug的解决方式

    今天小编就为大家分享一篇pytorch dataloader 取batch_size时候出现bug的解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-02-02
  • python实现web方式logview的方法

    python实现web方式logview的方法

    这篇文章主要介绍了python实现web方式logview的方法,涉及Python基于web模块操作Linux命令的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-08-08
  • Python迭代器与生成器及作用示例详解

    Python迭代器与生成器及作用示例详解

    Python生成器在内存管理、延迟计算、生成无限序列以及提高代码简洁性和可读性方面都具有重要作用,这篇文章主要介绍了Python迭代器与生成器示例详解,需要的朋友可以参考下
    2024-02-02

最新评论