測試tensorflow-gpu-2.0安裝成功

方法一 測試代碼以下:web 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
相關文章
相關標籤/搜索