查看当前TensorFlow版本
import tensorflow as tf
print(tf.__version__)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/ca94d035ae0000da1a660d0a8771852b.png)
查看当前主机上运行的设备
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')
print(gpus)
print(cpus)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/8a3c82c4ae529dc1a7551dc478db9a6f.png)
查看GPU是否可用,指定在CPU/GPU上运行
import tensorflow as tf
with tf.device('/cpu:0'):
cpu_a = tf.random.normal([10000, 1000])
cpu_b = tf.random.normal([1000, 2000])
cpu_c = tf.matmul(cpu_a, cpu_b)
print("cpu_a:", cpu_a.device)
print("cpu_b:", cpu_b.device)
print("cpu_c:", cpu_c.device)
print(tf.config.list_physical_devices('GPU'))
with tf.device('/gpu:0'):
gpu_a = tf.random.normal([10000, 1000])
gpu_b = tf.random.normal([1000, 2000])
gpu_c = tf.matmul(gpu_a, gpu_b)
print("gpu_a:", gpu_a.device)
print("gpu_b:", gpu_b.device)
print("gpu_c:", gpu_c.device)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/f92e85a7702612673826c2cf2e14cdf1.png)
比较在CPU和GPU上的运行时间
import tensorflow as tf
import timeit
def cpu_run():
with tf.device('/cpu:0'):
cpu_a = tf.random.normal([10000, 1000])
cpu_b = tf.random.normal([1000, 2000])
c = tf.matmul(cpu_a, cpu_b)
return c
def gpu_run():
with tf.device('/gpu:0'):
gpu_a = tf.random.normal([10000, 1000])
gpu_b = tf.random.normal([1000, 2000])
c = tf.matmul(gpu_a, gpu_b)
return c
cpu_time = timeit.timeit(cpu_run, number=10)
gpu_time = timeit.timeit(gpu_run, number=10)
print("cpu:", cpu_time, " gpu:", gpu_time)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/9a5a43c8ce1245dcdb3cf2d23401bff5.png)