tensoflow 使用CPU和GPU的准确代码，重点准确！

tensorflow 使用CPU和GPU,某度，某google一搜一大堆，排名靠前的都不一定对。如果你搜到我这篇，那么你肯定准确了。

前提：你安装了cuda等GPU环境没有问题

1 、使用CPU

很多人会说，cpu啥都不用配置，直接用就是cpu,那是因为你没安装好GPU的cuda驱动，如果安装好了，默认tensorflow是使用GPU的而且，乱占用。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""  #这句关键
import time

import tensorflow as tf

with tf.device('/cpu:0'):
    with tf.Session() as sess:
    # Here 6 GBs of GPU RAM are allocated.
    #     time.sleep(10)
        a = tf.constant([1,2,3,4],dtype=tf.float32)
        a = tf.expand_dims(a,-1)
        a_sum = tf.reduce_sum(a,1)
        print(sess.run(a_sum))

2、使用GPU

主要要可控呀，配好用某一个GPU就用某一个GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 用GPU 0号卡
import time

import tensorflow as tf

with tf.device('/gpu:0'):
    with tf.Session() as sess:
    # Here 6 GBs of GPU RAM are allocated.
#     time.sleep(10)
        a = tf.constant([1,2,3,4],dtype=tf.float32)
        a = tf.expand_dims(a,-1)
        a_sum = tf.reduce_sum(a,1)
        print(sess.run(a_sum))