可以从Java应用程序以GPU模式运行TensorFlow吗？

Question

我想使用Python的TensorFlow对象检测模块中的一个预先训练的模块来训练我自己的数据集，之后，我想用Java运行经过训练的数据集。我已经安装并测试了TensorFlow for Java。但它只在CPU上运行

当我在互联网上搜索时，用Java训练TensorFlow是不可能的，所以为了训练，我使用Python，然后为了运行，我想用Java，但问题是GPU，我不知道是否可以从Java运行

我的工作流程是这样的：

• 准备好数据集
• 使用一个预先训练的模块（来自TensorFlow）训练数据集 目标检测SSD_mobilenet_v1）
• 使用OpenCV从相机读取帧
• 通过TensorFlow使用我自己的火车数据集检测对象

Answer 1

Tensorflow for Java确实支持GPU

If your Linux system has an NVIDIA® GPU and your TensorFlow Java program requires GPU acceleration, then add the following to the project's pom.xml instead:

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>libtensorflow</artifactId>
  <version>1.5.0</version>
</dependency>
<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>libtensorflow_jni_gpu</artifactId>
  <version>1.5.0</version>
</dependency>

请注意，GPU仅在系统满足GPU support requirements时可用，此时仅在Linux系统上可用

（Installing TensorFlow for Java）

尽管GPU目前在Windows上不可用，但一期GPU support for Java in windows中的评论表明，他们正在积极开发GPU