如何确保我的 tensorFlow Java 程序在 Windows 上使用我的 GPU？

Question

所以我在 Windows 中为 CPU 实现了 Java tensorFlow。 现在 tf 1.10 之后的版本允许在 Windows 中进行 GPU 计算，我尝试建立连接。 现在在 maven pom 中，我调用 tensorflow_jni_gpu 而不是 tensorflow_jni。 程序没有失败，但在计算上似乎没有任何区别。 如何检查我的程序是否正在使用 GPU 或如何强制它使用它？

Answer 1

要使用 GPU，您应该在 POM 中有以下依赖项来处理 libtensorflow 而不是 tensorflow：`

<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>libtensorflow</artifactId>
    <version>1.15.0</version>
</dependency>

<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>libtensorflow_jni_gpu</artifactId>
    <version>1.15.0</version>
</dependency>

<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>proto</artifactId>
    <version>1.15.0</version>
</dependency>

`

以及以下代码：

byte[] config = ConfigProto.newBuilder()
            .setGpuOptions(GPUOptions.newBuilder().setAllowGrowth(true))
            .build().toByteArray();       
    final Runner runner = Session.runner();
    runner.setOptions(config);

关于知道它是在 CPU 还是 GPU 上运行，您应该在应用程序日志中看到它要么 tensorflow 尚未针对某些 CPU 指令构建，要么 tensorflow 设法加载 CUDA dll 等有关操作系统