[Перевод] Установите TensorFlow для Java

Google искусственный интеллект TensorFlow
[Перевод] Установите TensorFlow для Java

Эта статья взята изПрограмма перевода самородковОрганизуется переводОфициальная документация TensorFlow. Если вам интересно, добро пожаловатьПодать заявку на переводчика, изучив учебник переводчика, участвуйте в переводе и сопоставлении статей и документов. Мы также набираем переводчиков TensorFlow, поэтому, пожалуйста, примите активное участие.

TensorFlow предоставляет API для Java-программ. Эти API специально используются в приложениях Java для загрузки и выполнения моделей, созданных на Python. В этом руководстве объясняется, как установить и использовать его в приложении.TensorFlow for Java.

предупреждать:Java-API TensorFlowНетВходит в TensorFlow.Гарантия стабильности API

Платформа поддержки

TensorFlow для Java поддерживает работу в следующих операционных системах:

  • Linux
  • Mac OS X
  • Windows
  • Android

Инструкции по установке на Android находятся в отдельномСтраница поддержки Android TensorFlowсередина. После установки проверьте этот TensorFlow на AndroidПолный пример.

Использование TensorFlow в проекте Maven

Если в вашем проекте используетсяApache Maven, чтобы использовать TensorFlow Java API, в проектеpom.xmlДобавьте к нему следующее:

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow</artifactId>
  <version>1.4.0-rc0</version>
</dependency>

Пример

Например, эти шаги создадут проект Maven, использующий TensorFlow:

  1. создание проектаpom.xml:

    <project>
    <modelVersion>4.0.0</modelVersion>
    <groupId>org.myorg</groupId>
    <artifactId>hellotf</artifactId>
    <version>1.0-SNAPSHOT</version>
    <properties>
      <exec.mainClass>HelloTF</exec.mainClass>
      <!-- 这个样例代码至少需要 JDK 1.7 。 -->
      <!-- maven 编译器插件默认为一个更低的版本 -->
      <maven.compiler.source>1.7</maven.compiler.source>
      <maven.compiler.target>1.7</maven.compiler.target>
    </properties>
    <dependencies>
      <dependency>
        <groupId>org.tensorflow</groupId>
        <artifactId>tensorflow</artifactId>
        <version>1.4.0-rc0</version>
      </dependency>
    </dependencies>
</project>

  2. Создайте исходный файл (src/main/java/HelloTF.java):

    import org.tensorflow.Graph; import org.tensorflow.Session; import org.tensorflow.Tensor; import org.tensorflow.TensorFlow;

    public class HelloTF { public static void main(String[] args) throws Exception { try (Graph g = new Graph()) { final String value = "Hello from " + TensorFlow.version();

         // 使用一个简单操作、一个名为 "MyConst" 的常数和一个值 "value" 来构建计算图。
     try (Tensor t = Tensor.create(value.getBytes("UTF-8"))) {
       // Java API 目前还不包含足够方便的函数来执行“加”操作。
       g.opBuilder("Const", "MyConst").setAttr("dtype", t.dataType()).setAttr("value", t).build();
     }

     // 在一个 Session 中执行 "MyConst" 操作。
     try (Session s = new Session(g);
          Tensor output = s.runner().fetch("MyConst").run().get(0)) {
       System.out.println(new String(output.bytesValue(), "UTF-8"));
     }
   }
 }

    }

  3. Скомпилируйте и выполните:

     # 使用 -q 来隐藏 mvn 工具的日志
mvn -q compile exec:java

Предыдущая команда должна вывестиHello from version. Если выходные данные успешны, вы успешно установили TensorFlow для Java и можете использовать его в проекте Maven. В случае неудачи перейдите кStack OverflowПоиск возможных решений. Вы можете пропустить чтение остальной части этого документа.

Использование TensorFlow под JDK

В этом разделе описывается, как использовать TensorFlow с помощью команд java и javac, установленных вместе с JDK. Если вы используете Apache Maven в своем проекте, обратитесь к более простому методу установки, используя предыдущий раздел.

Установить на Linux или Mac OS

Выполните следующие шаги, чтобы установить TensorFlow для Java в Linux или Mac OS:

  1. скачатьlibtensorflow.jar, Это Java-архив TensorFlow (JAR).

  2. Решите, хотите ли вы запускать TensorFlow для Java только на ЦП или с помощью графического процессора. Чтобы помочь вам принять решение, прочитайте раздел «Выбор TensorFlow для установки» в следующем руководстве:

    • @{install_linux#determine_which_tensorflow_to_installУстановите TensorFlow в Linux}
    • @{install_mac#determine_which_tensorflow_to_installУстановите TensorFlow в Mac OS}

  3. Поддержите свою операционную систему и процессор, выполнив следующие команды оболочки для загрузки и извлечения соответствующих файлов Java Native Interface (JNI):

    TF_TYPE="cpu" # 默认处理器是 CPU 。如果你想要使用 GPU ，就将它设置成 "gpu" 。
OS=$(uname -s | tr '[:upper:]' '[:lower:]')
mkdir -p ./jni
curl -L \
  "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-${TF_TYPE}-${OS}-x86_64-1.4.0-rc0.tar.gz" |
  tar -xz -C ./jni

Установить на Windows

Установите TensorFlow для Java в Windows, выполнив следующие действия:

  1. скачатьlibtensorflow.jar, Это Java-архив TensorFlow (JAR).
  2. Скачать для TensorFlow для Java в WindowsФайлы собственного интерфейса Java (JNI).
  3. Разархивируйте этот ZIP-файл.

Проверить установку

После установки TensorFlow для Java вHelloTF.javaВведите следующий код в файл, чтобы проверить установку:

import org.tensorflow.Graph;
import org.tensorflow.Session;
import org.tensorflow.Tensor;
import org.tensorflow.TensorFlow;

public class HelloTF {
  public static void main(String[] args) throws Exception {
    try (Graph g = new Graph()) {
      final String value = "Hello from " + TensorFlow.version();

      // 使用一个简单操作、一个名为 "MyConst" 的常数和一个值 "value" 来构建计算图。
      try (Tensor t = Tensor.create(value.getBytes("UTF-8"))) {
        // Java API 目前还不包含足够方便的函数来执行“加”操作。
        g.opBuilder("Const", "MyConst").setAttr("dtype", t.dataType()).setAttr("value", t).build();
      }

      // 在一个 Session 中执行 "MyConst" 操作。
      try (Session s = new Session(g);
           Tensor output = s.runner().fetch("MyConst").run().get(0)) {
        System.out.println(new String(output.bytesValue(), "UTF-8"));
      }
    }
  }
}

и используйте следующие команды для компиляции и запускаHelloTF.java.

компилировать

При компиляции Java-программы, использующей TensorFlow, загруженный.jarФайл должен быть у васclasspathсередина. Например, вы можете использовать флаги компиляции с помощью такой директивы, как-cpбудет загружен.jarфайл, включенный в вашclasspathсередина:

javac -cp libtensorflow-1.4.0-rc0.jar HelloTF.java

бегать

Чтобы запустить программу Java, зависящую от TensorFlow, убедитесь в следующем: Для JVM доступны два файла:

  • загружен.jarдокумент
  • Извлеченная библиотека JNI

Например, используйте следующую команду для запуска в Linux и Mac OS X.HelloTFпрограмма:

java -cp libtensorflow-1.4.0-rc0.jar:. -Djava.library.path=./jni HelloTF

Используйте следующую команду для запуска в WindowsHelloTFпрограмма:

java -cp libtensorflow-1.4.0-rc0.jar;. -Djava.library.path=jni HelloTF

Если программа выводитHello from version, что означает, что вы успешно установили TensorFlow для Java и можете использовать API. Если программа выводит что-то другое, см.Stack Overflowнайти решение.

Расширенный пример

Более сложный пример см.LabelImage.java, который может идентифицировать объекты на изображениях.

сборка из исходников

TensorFlow имеет открытый исходный код. можно по этомуотдельный документИнструкции в Compile TensorFlow для Java из исходного кода.

Программа перевода самородковэто сообщество, которое переводит высококачественные технические статьи из Интернета сНаггетсДелитесь статьями на английском языке на . Охват контентаAndroid,iOS,внешний интерфейс,задняя часть,блокчейн,продукт,дизайн,искусственный интеллектЕсли вы хотите видеть более качественные переводы, пожалуйста, продолжайте обращать вниманиеПрограмма перевода самородков,официальный Вейбо,Знай колонку.

Категории