在前面文章中《Ubuntu16.04+GTX1050ti+CUDA8.0+TensorFlow-gpu+keras配置深度学习环境》已经配置好了系统。现在来进行GPU加速测试。
- 测试1:系统自动分配设备
#-*- coding:utf-8 -*-
import tensorflow as tf
# 新建一个 graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0,