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
相關文章
相關標籤/搜索