java - GPU 上的 DL4J 示例不起作用（用于学校项目）

Question

我已经有了深度学习的经验。在训练模型时，我使用 python 和 cuda 可以很好地与我的 GPU 配合使用。但是，现在我需要使用 Java （对于一个可选的学校项目，我想创建一个强化学习 AI，我必须使用 Java）。我对 Java 完全陌生，所以我关注了这个基于 DL4J 网站快速入门指南的视频。下载示例时，一切正常：image。

当我在 cpu（使用神经网络）上运行示例时，一切正常。但是，当我尝试这个“MultiGpuLenetMnistExample” （它在“dl4j-cuda-specific-examples”文件夹中）时，我收到以下错误，我试图寻找答案，但我没有找到我想要的（或者也许没看懂答案）。我猜这个问题来自 nd4j 后端或 jcublas 的东西，但我不知道该怎么做。

考虑到我对 Java 的所有细节还不太满意，当我寻找有相同问题的人时，我不明白他们在说什么，我只是发现了 pom.xml 文件的概念，例如……但是我看到回答的人通常要求 java -version、mvn --version 或 nvcc --version 所以他们在这里。

标签： javagpudeeplearning4jdl4j

Answer 1

根据您的屏幕截图，您正在尝试使用 CUDA 10.2 的 deps 运行项目，但您安装了 CUDA 10.0。将您的依赖项更改为 nd4j-cuda-10.0 而不是 nd4j-cuda-10.2