安裝

這裡使用 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個數

對於上面用到的手寫識別例子來說，需要修改檔案 /usr/lib/python2.7/site-

packages/tensorflow/models/image/mnist/convolutional.py 中建立 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:

修改完成後，使用環境變數 CPU_NUM 來指定需要使用的 CPU 個數，然後再次執行手寫識別例子

$ export CPU_NUM=2
$ python -m tensorflow.models.image.mnist.convolutional

執行後，使用 top 命令來檢視程式的 CPU 使用情況。

以上這篇Tensorflow限制CPU個數例項就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。