Как запустить приложение Java с графическим процессором Tensorflow на Windows 10? - PullRequest
Новая
       17

Как запустить приложение Java с графическим процессором Tensorflow на Windows 10?

0 голосов
/ 17 июня 2019

Я пытался запустить Java-приложение Tensflow Flow на Windows 10. Я не смог запустить приложение на GPU каждый раз при запуске на CPU.

Я следовал инструкциям https://www.tensorflow.org/install/lang_java и https://www.tensorflow.org/install/gpu с установкой драйверов, инструментарием CUDA, cuDNN. Я проверил, все ли зависимости были в classpath с помощью обходчика зависимостей.

Зависимости Maven: 

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



    byte[] config =
            ConfigProto.newBuilder()
                    .setLogDevicePlacement(true)
                    .setAllowSoftPlacement(true)
                    .build()
                    .toByteArray();

во время выполнения:

I tenorflow / core / platform / cpu_feature_guard.cc: 141] Ваш ЦП поддерживает инструкции, которые этот двоичный файл TensorFlow не был скомпилирован для использования: AVX2

Я ожидаю запустить на GPU

Вы когда-нибудь сталкивались с этой проблемой? Любые хиты будут полезны, спасибо.

...