Tensorflow的GPU使用

一般情况，安装了 tensorflow的gpu版本即可自动运行，当然前提是配置好了环境．
只要安装过caffe基本就没问题．
安装：

pip install tensorflow-gpu

基本使用方法：

1 import os
2 os.environ['CUDA_VISIBLE_DEVICES'] = '0, 1'  
3 
4 gpuConfig = tf.ConfigProto(allow_soft_placement=True)
5 gpuConfig.gpu_options.allow_growth = True  
6 
7 with tf.Session(config=gpuConfig) as sess:
8      pass

查看是否使用：

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))