而一台机器上不同GPU的名称是不同的，第n个GPU在TensorFlow中的名称为/gpu:n。比如第一个GPU的名称为/gpu:0，第二个GPU名称为/gpu:1，以此类推。
这个还属可以把运算放在不同的设备上。
with tf.device('/cpu:0'):
a= tf.constant([1.0, 2.0, 3.0], shape=[3], name='a')
b= tf.constant([1.0, 2.0, 3.0], shape=[3], name='b')
with tf.device('/gpu:1'):
c= a + b
而一台机器上不同GPU的名称是不同的，第n个GPU在TensorFlow中的名称为/gpu:n。比如第一个GPU的...
with tf.Session() as sess: with tf.device("/gpu:0"): ...
1、关于TensorFlow 相关网站：TensorFlow、TensorFlow中文社区 TensorFlow™...
本章节目录 Tensorflow简介Tensorflow相关网站 Tensorflow的安装Tensorflow学...
离线安装 Tensorflow docker pull tensorflow/tensorflow docker ...
Hello TensorFlow TensorFlow C library TensorFlow Go bin...
Tensorflow Intro 1. Tensorflow core concept Tensorflow is...
下载TensorFlow https://github.com/tensorflow/tensorflow/tre...
在最新 TensorFlow 中， TensorFlow Serving 是 TensorFlow Extende...
目录 TensorFlow简介 TensorFlow基本概念 Using TensorFlow Optimizat...
本文标题：tensorflow tf.device
本文链接：https://www.haomeiwen.com/subject/pntnoftx.html
网友评论