運(yùn)行tensorflow python程序,限制對(duì)GPU和CPU的占用操作

更新時(shí)間：2020年02月06日 09:48:35 作者：keep_forward

今天小編就為大家分享一篇運(yùn)行tensorflow python程序,限制對(duì)GPU和CPU的占用操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

一般情況下，運(yùn)行tensorflow時(shí)，默認(rèn)會(huì)占用可以看見(jiàn)的所有GPU，那么就會(huì)導(dǎo)致其它用戶或程序無(wú)GPU可用，那么就需要限制程序?qū)PU的占用。并且，一般我們的程序也用不了所有的GPU資源，只是強(qiáng)行霸占著，大部分資源都不會(huì)用到，也不會(huì)提升運(yùn)行速度。

使用nvidia-smi可以查看本機(jī)的GPU使用情況，如下圖，這里可以看出，本機(jī)的GPU型號(hào)是K80，共有兩個(gè)K80，四塊可用（一個(gè)K80包括兩塊K40）。

1、如果是只需要用某一塊或某幾塊GPU，可以在運(yùn)行程序時(shí)，利用如下命令運(yùn)行：CUDA_VISIBLE_DEVICES=0,1 python test.py

這里表示只有GPU 0和1對(duì)程序可見(jiàn)，因此也就限制了程序只能用GPU 0和1

同樣，也可以在代碼里指定

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

如果想只用CPU，不用CPU來(lái)運(yùn)行程序，可以用如下命令(所有GPU都不可見(jiàn))：

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:

前面是對(duì)GPU的限制，那如果不用GPU，只用CPU呢？如何限制對(duì)CPU的使用呢？

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

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:

以上這篇運(yùn)行tensorflow python程序,限制對(duì)GPU和CPU的占用操作就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python 實(shí)現(xiàn)矩陣按對(duì)角線打印
今天小編就為大家分享一篇python 實(shí)現(xiàn)矩陣按對(duì)角線打印，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-11-11
Python中關(guān)于列表的常規(guī)操作范例以及介紹
列表是一種有序的集合，可以隨時(shí)添加和刪除其中的元素。在python中使用的頻率非常高，本篇文章對(duì)大家的學(xué)習(xí)或工作具有一定的價(jià)值，需要的朋友可以參考下
2021-09-09
在Pycharm中安裝Pandas庫(kù)方法(簡(jiǎn)單易懂)
這篇文章主要介紹了在Pycharm中安裝Pandas庫(kù)方法,文中通過(guò)圖文介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2021-02-02
kNN算法python實(shí)現(xiàn)和簡(jiǎn)單數(shù)字識(shí)別的方法
這篇文章主要介紹了kNN算法python實(shí)現(xiàn)和簡(jiǎn)單數(shù)字識(shí)別的方法,詳細(xì)講述了kNN算法的優(yōu)缺點(diǎn)及原理,并給出了應(yīng)用實(shí)例,需要的朋友可以參考下
2014-11-11
淺析Python編寫函數(shù)裝飾器
這篇文章主要介紹了Python編寫函數(shù)裝飾器的相關(guān)資料,需要的朋友可以參考下
2016-03-03
教你如何在pycharm中使用less
這篇文章主要介紹了如何在pycharm中使用less,操作步驟真的很簡(jiǎn)單，本文通過(guò)圖文并茂的形式給大家詳細(xì)介紹，需要的朋友可以參考下
2021-10-10
Python中字典(dict)和列表(list)的排序方法實(shí)例
這篇文章主要介紹了Python中字典(dict)和列表(list)的排序方法實(shí)例,總結(jié)來(lái)說(shuō)優(yōu)先使用內(nèi)置的sort()方法進(jìn)行排序,需要的朋友可以參考下
2014-06-06
pytorch dataloader 取batch_size時(shí)候出現(xiàn)bug的解決方式
今天小編就為大家分享一篇pytorch dataloader 取batch_size時(shí)候出現(xiàn)bug的解決方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2020-02-02
python實(shí)現(xiàn)web方式logview的方法
這篇文章主要介紹了python實(shí)現(xiàn)web方式logview的方法,涉及Python基于web模塊操作Linux命令的技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下
2015-08-08
Python迭代器與生成器及作用示例詳解
Python生成器在內(nèi)存管理、延遲計(jì)算、生成無(wú)限序列以及提高代碼簡(jiǎn)潔性和可讀性方面都具有重要作用,這篇文章主要介紹了Python迭代器與生成器示例詳解,需要的朋友可以參考下
2024-02-02