测试tensorflow-gpu-2.0安装成功
方法一测试代码如下:import tensorflow as tfimport timeitwith tf.device('/cpu:0'): cpu_a = tf.random.normal([10000, 1000]) cpu_b = tf.random.normal([1000, 2000]) print(cpu_a.device, cpu_b.device)with ...
原创
2019-08-20 12:03:24 ·
29492 阅读 ·
14 评论