构建 NeuralNetConfiguration 配置类时出现 deeplearning4j 错误：org.nd4j.linalg.factory.Nd4jBackend$NoAvailableBackendException

Question

我正在尝试在 intellij 中启动一个新项目，以使用 deeplearning4j 进行一些深度强化学习，但在构建 NeuralNetConfiguration 类时出现错误。我认为问题出在我的 pom.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 ?在这方面我有点新手......

<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 的本机实现作为 CPU 后端，请添加以下依赖项。

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

参考这个documentation。

我希望这会有所帮助。