Tensorflow 使用GPUs

Tensorflow 支援使用 CPU 和 GPU 做運算

  • "/cpu:0": The CPU of your machine.
  • "/gpu:0": The GPU of your machine, if you have one.
  • "/gpu:1": The second GPU of your machine, etc.

用 with tf.device() 來分配這個語句下使用的設備

TF_GPUs1

log.device_placement = True 紀錄我們使用device的情況

allow_soft_placement = True 避免指定的device不存在,讓他能自行分配到存在且可運行的地方

TF_GPUs2

我沒有多顆CPU,其他的語法先不試



發表留言