我在dPointsWS中的设备内存中有一个点云,其中的内存布局首先是x-,然后是所有的y-,最后是所有的z坐标。我用推力来计算这个点云的紧轴对齐包围盒(AABB)。thrust::device_ptr<Real>, thrust::device_ptr<Real>> thrustAABB[3];
// do parrallel min_max reduction on GPU我已经很清楚地下载了结果到主机内存在最后,但我想避免这种情况。我找到了以下文章:。由于约简函数返回
我怀疑它使用的是GPU,所以我做了个测试。我修改了一个示例来强制使用图形处理器。logits = mnist.inference(images_placeholder, FLAGS.hidden1, FLAGS.hidden2)
+ with tf.device("/gpua device to node 'softmax_linear/biases': Could not satisfy explicit device specification '/device:GPU<