安裝

這里使用 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個數實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。