ошибка deeplearning4j при создании класса конфигурации NeuralNetConfiguration: org.nd4j.linalg.factory.Nd4jBackend $ NoAvailableBackendException

Question

Я пытаюсь начать новый проект в intellij, чтобы провести глубокое обучение с подкреплением с deeplearning4j, но получаю ошибки при построении класса NeuralNetConfiguration. Я думаю, что проблема с моим файлом. xml.

Exception in thread "main" java.lang.ExceptionInInitializerError
at org.deeplearning4j.nn.conf.MultiLayerConfiguration$Builder.build(MultiLayerConfiguration.java:719)
at org.deeplearning4j.nn.conf.NeuralNetConfiguration$ListBuilder.build(NeuralNetConfiguration.java:258)
at test.<clinit>(test.java:20)
Caused by: java.lang.RuntimeException: org.nd4j.linalg.factory.Nd4jBackend$NoAvailableBackendException: Please ensure that you have an nd4j backend on your classpath. Please see: http://nd4j.org/getstarted.html
at org.nd4j.linalg.factory.Nd4j.initContext(Nd4j.java:5131)
at org.nd4j.linalg.factory.Nd4j.<clinit>(Nd4j.java:226)
... 3 more

У меня есть обе эти зависимости в моем файле pom. xml, который, как я думал, будет достаточно, ср. http://nd4j.org/getstarted.html? Я немного новичок ie, когда дело доходит до всего этого ...

<dependencies>
    <dependency>
        <groupId>org.nd4j</groupId>
        <artifactId>nd4j-native-platform</artifactId>
        <version>1.0.0-beta6</version>
    </dependency>

    <dependency>
        <groupId>org.deeplearning4j</groupId>
        <artifactId>deeplearning4j-core</artifactId>
        <version>1.0.0-beta6</version>
    </dependency>
</dependencies>

Answer 1

Если вы используете установленное CUDA v9.2 + и совместимое с NVIDIA оборудование, добавьте приведенную ниже зависимость.

<dependency>
 <groupId>org.nd4j</groupId>
 <artifactId>nd4j-cuda-10.1</artifactId>
 <version>1.0.0-beta6</version>
</dependency>

если вы используете собственную реализацию ND4J в качестве бэкэнда ЦП, добавьте следующую зависимость.

 <dependency>
    <groupId>org.nd4j</groupId>
    <artifactId>nd4j-native</artifactId>
    <version>1.0.0-beta6</version>
</dependency>

см. Эту документацию .

Надеюсь, это поможет.