Tensorflow限制CPU個數(shù)實例
更新時間：2020年02月06日 15:00:50 作者：kongxx
今天小編就為大家分享一篇Tensorflow限制CPU個數(shù)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
安裝
這里使用 Pip 來安裝 Tensorflow CPU 版
$ sudo pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
安裝完成后運行庫中自帶的手寫識別例子來檢查安裝是否成功
$ cd /usr/lib/python2.7/site-packages/tensorflow/models/image/mnist $ python convolutional.py ...
或者運行
$ python -m tensorflow.models.image.mnist.convolutional ...
限制CPU個數(shù)
對于上面用到的手寫識別例子來說，需要修改文件 /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 來指定需要使用的 CPU 個數(shù)，然后再次運行手寫識別例子
$ export CPU_NUM=2 $ python -m tensorflow.models.image.mnist.convolutional
運行后，使用 top 命令來查看程序的 CPU 使用情況。
以上這篇Tensorflow限制CPU個數(shù)實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
本文將介紹如何在Python中進行Redis操作，包括連接Redis、數(shù)據(jù)存儲、數(shù)據(jù)檢索和其他常見操作,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-11-11
今天小編就為大家分享一篇TensorBoard 計算圖的可視化實現(xiàn)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要介紹了python Flask 裝飾器順序問題解決，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-08-08
這篇文章主要介紹了 Python 元類實例解析,此文的主要任務就是給大家徹底講明白什么是元類，需要的朋友可以參考下2018-04-04
python django框架中使用FastDFS分布式文件系統(tǒng)的安裝方法
這篇文章主要介紹了python-django框架中使用FastDFS分布式文件系統(tǒng)的安裝方法，本文圖文并茂給大家介紹的非常詳細，具有一定的參考借鑒價值 ,需要的朋友可以參考下2019-06-06![Python中的shape[0]、shape[1]和shape[-1]使用方法](http://img.jbzj.com/images/xgimg/bcimg5.png)
Python中的shape[0]、shape[1]和shape[-1]使用方法
shape函數(shù)是Numpy中的函數(shù)，它的功能是讀取矩陣的長度，比如shape[0]就是讀取矩陣第一維度的長度，這篇文章主要介紹了Python中的shape[0]、shape[1]和shape[-1]使用方法,需要的朋友可以參考下2023-07-07
Django Auth 應用一般用在用戶的登錄注冊上，用于判斷當前的用戶是否合法。本文將介紹Auth的另一個功能，即認證用戶身份，感興趣的同學可以關注一下2021-12-12
零基礎寫python爬蟲之使用urllib2組件抓取網頁內容
文章詳細介紹了在python2.5環(huán)境下，如何使用urllib2這個python自帶的組件進行抓取指定網頁內容的，整個過程記錄的非常的詳細，也很簡單，有需要的朋友可以參考下，寫出自己的python爬蟲2014-11-11
這篇文章主要介紹了利用Python實現(xiàn)kNN算法的代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-08-08
這篇文章主要介紹了Python守護進程(daemon)代碼實例,本文直接給出實現(xiàn)代碼,代碼中包含詳細注釋,需要的朋友可以參考下2015-03-03