Tensorflow-gpu2.0.0安装以及tensorflow-gpu安装成功的测试程序

本来: 测试代码:python import tensorflow as tf import timeit with 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) wi
相关文章
相关标签/搜索