运行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正则表达式常见面试题，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-01-01
python返回数组索引的方法实现
本文介绍了三种在Python中返回数组索引的方法,主要包括index()方法,enumerate()方法及使用numpy库的argwhere()函数,具有一定的参考价值,感兴趣的可以了解一下
2024-01-01
python 3.0 模拟用户登录功能并实现三次错误锁定
Python的3.0版本，常被称为Python 3000，或简称Py3k。这篇文章主要介绍了python 3.0 模拟用户登录功能并实现三次错误锁定,需要的朋友可以参考下
2017-11-11
python递归函数使用详解
递归函数是一种在函数内部调用自身的编程技巧。在Python中，我们可以使用递归函数来解决一些需要重复执行相同操作的问题。递归函数通常包含两个部分：基本情况和递归情况。基本情况是指函数停止调用自身的条件，而递归情况是指函数调用自身来解决更小规模的问题。
2023-09-09
基于循环神经网络(RNN)实现影评情感分类
这篇文章主要为大家详细介绍了基于循环神经网络(RNN)实现影评情感分类，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-03-03
python实现图像随机裁剪的示例代码
这篇文章主要介绍了python实现图像随机裁剪的示例代码，帮助大家更好的理解和使用python处理图片，感兴趣的朋友可以了解下
2020-12-12
python and or用法详解
这篇文章主要介绍了python and or用法详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-06-06
Python执行时间的几种计算方法
这篇文章主要介绍了Python执行时间的几种计算方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-07-07
python __init__与 __new__的区别
本文主要介绍了python __init__与 __new__的区别，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-02-02
Python try except else使用详解
这篇文章主要介绍了Python try except else使用详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-01-01