java - 在 GPU Java 上配置 Tensorflow

Question

如何使用 Java API 为 Tensorflow 设置以下配置“gpu_options.allow_growth = True”？

我这样试过:

model.session().runner()
                .setOptions("gpu_options.allow_growth = True".getBytes())
                .feed("image_tensor", input).fetch("detection_scores")
                .fetch("detection_classes").fetch("detection_boxes").fetch("num_detections").run();

我收到以下错误:Unparseable RunOptions proto。

Answer 1

我认为这是配置的方式:

      ConfigProto config = ConfigProto.newBuilder()
                .setGpuOptions(GPUOptions.newBuilder().setAllowGrowth(true))
                .build();

      model.session().runner()
                    .setOptions(config.toByteArray())
                    .feed("image_tensor", input).fetch("detection_scores")
                    .fetch("detection_classes").fetch("detection_boxes").fetch("num_detections").run();