Using GPU with TensorFlow: Specifying Device with tf.device Function
TensorFlow programs can specify the device for each operation using the tf.device function. This device can be a local CPU or GPU, or it can be a remote server. TensorFlow assigns a name to each available device, and the tf.device function can specify the device for executing operations using the device’s name. For example, the … Read more