TensorFlow之設備(device)詳解

對TensorFlow不瞭解的,猛的一看還以爲TensorFlow是個設備什麼的,其實本文要說的是Python的TensorFlow模塊下一個叫device()的方法,即: import tensorflow as tf with tf.device('/gpu:0'): ... ... ... 設備是指一塊可以用來運算並且擁有自己的地址空間的硬件,如CPU和GPU Te
相關文章
相關標籤/搜索