使用CUDA运行Dl4j时出现问题

Question

我正在尝试使用GPU而不是CPU运行dl4j模型。该模型使用CPU运行良好。所以我决定尝试CUDA来利用我的GPU。我遵循了这里描述的每个步骤，对于CUDA安装，我按照NVIDIA的说明从这里安装CUDA工具包。代码编译正常，但出现错误：

Skipped [JCublasBackend] backend (unavailable): java.lang.UnsatisfiedLinkError: C:\Users\albertb\.javacpp\cache\cuda-10.2-7.6-1.5.3-windows-x86_64.jar\org\bytedeco\cuda\windows-x86_64\jnicudart.dll: Can't find dependent libraries
Exception in thread "main" java.lang.ExceptionInInitializerError
    at TestCUDA.run(TestCUDA.java:12)
    at TestCUDA.main(TestCUDA.java:7)
Caused by: java.lang.RuntimeException: org.nd4j.linalg.factory.Nd4jBackend$NoAvailableBackendException: Please ensure that you have an nd4j back-end on your classpath. Please see: https://deeplearning4j.konduit.ai/nd4j/backend
    at org.nd4j.linalg.factory.Nd4j.initContext(Nd4j.java:5094)
    at org.nd4j.linalg.factory.Nd4j.<clinit>(Nd4j.java:270)
    ... 2 more
Caused by: org.nd4j.linalg.factory.Nd4jBackend$NoAvailableBackendException: Please ensure that you have an nd4j backend on your classpath. Please see: https://deeplearning4j.konduit.ai/nd4j/backend
    at org.nd4j.linalg.factory.Nd4jBackend.load(Nd4jBackend.java:221)
    at org.nd4j.linalg.factory.Nd4j.initContext(Nd4j.java:5091)
    ... 3 more

我的项目是使用IntelliJ上的Maven创建的。所有的依赖项都是可以的，否则代码将无法编译。必须缺少某些后端依赖项、库或安装。可能是版本不兼容的问题
消息请确保您的类路径上有一个nd4j后端为我提供了一些关于必须存在的后端jar的提示。但是Maven应该负责这件事。。。据我所知。

我在网上搜索了一些解决方案或示例或教程...任何东西！！上面链接中有相同指令的所有冒号。

一些技术信息：

• Windows 10

我会继续努力解决这个问题，但我会感谢任何帮助。谢谢

Answer 1

我决定了！现在，我的例子在GeForce3090上工作！！！

我POM的一部分：

 org.example
    dl4jCuda
    1.0-SNAPSHOT

        <dependency>
            <groupId>org.deeplearning4j</groupId>
            <artifactId>deeplearning4j-cuda-11.0</artifactId>
            <version>1.0.0-SNAPSHOT</version>
        </dependency>
        <dependency>
            <groupId>org.deeplearning4j</groupId>
            <artifactId>deeplearning4j-core</artifactId>
            <version>1.0.0-SNAPSHOT</version>
        </dependency>

您还需要从源代码构建nd4j-cuda-11.0，并将所有*jar和dll文件添加到您的项目中！

Answer 2

您可能安装了错误的cuda版本。nd4j cuda版本在工件id中指定，如：nd4j-cuda-11.0/10.2。

如果需要，可以使用javacpp的redist工件：https://search.maven.org/artifact/org.bytedeco/cuda-platform-redist/11.0-8.0-1.5.4/jar

这是cuda 11的最新版本。Nd4j只是在其所有本机互操作中使用javacpp。