Tensorflow限制CPU個(gè)數(shù)實(shí)例
安裝
這里使用 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)文章
TensorBoard 計(jì)算圖的可視化實(shí)現(xiàn)
今天小編就為大家分享一篇TensorBoard 計(jì)算圖的可視化實(shí)現(xiàn),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2020-02-02python django框架中使用FastDFS分布式文件系統(tǒng)的安裝方法
這篇文章主要介紹了python-django框架中使用FastDFS分布式文件系統(tǒng)的安裝方法,本文圖文并茂給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值 ,需要的朋友可以參考下2019-06-06Python中的shape[0]、shape[1]和shape[-1]使用方法
shape函數(shù)是Numpy中的函數(shù),它的功能是讀取矩陣的長(zhǎng)度,比如shape[0]就是讀取矩陣第一維度的長(zhǎng)度,這篇文章主要介紹了Python中的shape[0]、shape[1]和shape[-1]使用方法,需要的朋友可以參考下2023-07-07Django Auth應(yīng)用實(shí)現(xiàn)用戶身份認(rèn)證
Django Auth 應(yīng)用一般用在用戶的登錄注冊(cè)上,用于判斷當(dāng)前的用戶是否合法。本文將介紹Auth的另一個(gè)功能,即認(rèn)證用戶身份,感興趣的同學(xué)可以關(guān)注一下2021-12-12零基礎(chǔ)寫python爬蟲之使用urllib2組件抓取網(wǎng)頁(yè)內(nèi)容
文章詳細(xì)介紹了在python2.5環(huán)境下,如何使用urllib2這個(gè)python自帶的組件進(jìn)行抓取指定網(wǎng)頁(yè)內(nèi)容的,整個(gè)過程記錄的非常的詳細(xì),也很簡(jiǎn)單,有需要的朋友可以參考下,寫出自己的python爬蟲2014-11-11Python守護(hù)進(jìn)程(daemon)代碼實(shí)例
這篇文章主要介紹了Python守護(hù)進(jìn)程(daemon)代碼實(shí)例,本文直接給出實(shí)現(xiàn)代碼,代碼中包含詳細(xì)注釋,需要的朋友可以參考下2015-03-03