Tensorflow限制CPU個(gè)數(shù)實(shí)例
更新時(shí)間：2020年02月06日 15:00:50 作者：kongxx
今天小編就為大家分享一篇Tensorflow限制CPU個(gè)數(shù)實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
安裝
這里使用 Pip 來(lái)安裝 Tensorflow CPU 版
$ sudo pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
安裝完成后運(yùn)行庫(kù)中自帶的手寫識(shí)別例子來(lái)檢查安裝是否成功
$ cd /usr/lib/python2.7/site-packages/tensorflow/models/image/mnist $ python convolutional.py ...
或者運(yùn)行
$ python -m tensorflow.models.image.mnist.convolutional ...
限制CPU個(gè)數(shù)
對(duì)于上面用到的手寫識(shí)別例子來(lái)說(shuō)，需要修改文件 /usr/lib/python2.7/site-
packages/tensorflow/models/image/mnist/convolutional.py 中創(chuàng)建 Session 部分
修改前
with tf.Session(config=config) as s:
修改后
cpu_num = int(os.environ.get('CPU_NUM', 1))
config = tf.ConfigProto(device_count={"CPU": cpu_num},
inter_op_parallelism_threads = cpu_num,
intra_op_parallelism_threads = cpu_num,
log_device_placement=True)
with tf.Session(config=config) as s:
修改完成后，使用環(huán)境變量 CPU_NUM 來(lái)指定需要使用的 CPU 個(gè)數(shù)，然后再次運(yùn)行手寫識(shí)別例子
$ export CPU_NUM=2 $ python -m tensorflow.models.image.mnist.convolutional
運(yùn)行后，使用 top 命令來(lái)查看程序的 CPU 使用情況。
以上這篇Tensorflow限制CPU個(gè)數(shù)實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
在Python的函數(shù)中，我們將其分為內(nèi)置函數(shù)、自定義函數(shù)、main函數(shù)三個(gè)模塊，當(dāng)然，使用的過(guò)程中會(huì)涉及到變量以及參數(shù)，這些都會(huì)舉例進(jìn)行說(shuō)明，對(duì)Python函數(shù)使用相關(guān)知識(shí)感興趣的朋友跟隨小編一起看看吧2021-12-12
Kears 使用:通過(guò)回調(diào)函數(shù)保存最佳準(zhǔn)確率下的模型操作
這篇文章主要介紹了Kears 使用:通過(guò)回調(diào)函數(shù)保存最佳準(zhǔn)確率下的模型操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06
Python實(shí)現(xiàn)將照片變成卡通圖片的方法【基于opencv】
這篇文章主要介紹了Python實(shí)現(xiàn)將照片變成卡通圖片的方法,涉及Python基于opencv庫(kù)進(jìn)行圖片處理的相關(guān)操作技巧,需要的朋友可以參考下2018-01-01
ubuntu17.4下為python和python3裝上pip的方法
今天小編就為大家分享一篇ubuntu17.4下為python和python3裝上pip的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-06-06
本文主要介紹了NumPy內(nèi)存布局的實(shí)現(xiàn),括連續(xù)內(nèi)存布局（C順序）和分散內(nèi)存布局（Fortran順序）,并通過(guò)實(shí)例演示如何操作數(shù)組的內(nèi)存布局,感興趣的可以了解一下2024-01-01
Queue隊(duì)列中join()與task_done()的關(guān)系及說(shuō)明
這篇文章主要介紹了Queue隊(duì)列中join()與task_done()的關(guān)系及說(shuō)明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-02-02
本文主要介紹了Python腳本讀取Consul配置信息的方法示例，文中通過(guò)示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-11-11
這篇文章主要介紹了Pygame做一期吃豆子游戲的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
python中強(qiáng)制關(guān)閉線程與協(xié)程與進(jìn)程方法
python使用中多線程、多進(jìn)程、多協(xié)程使用是比較常見的。那么如果在多線程等的使用，我們這個(gè)時(shí)候我們想從外部強(qiáng)制殺掉該線程請(qǐng)問(wèn)如何操作？這篇文章帶你介紹，感興趣的同學(xué)可以參考閱讀2023-03-03
使用Pycharm(Python工具)新建項(xiàng)目及創(chuàng)建Python文件的教程
這篇文章主要介紹了使用Pycharm(Python工具)新建項(xiàng)目及創(chuàng)建Python文件的教程,本文通過(guò)圖文并茂的形式給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-04-04