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

【问题标题】：DL4J example on GPU doesn't work (for a school project)GPU 上的 DL4J 示例不起作用（用于学校项目）
【发布时间】：2020-01-01 16:45:08
【问题描述】：

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

当我在 cpu（使用神经网络）上运行示例时，一切正常。但是，当我尝试这个“MultiGpuLenetMnistExample”（它在“dl4j-cuda-specific-examples”文件夹中）时，我得到了以下error，我试图寻找答案，但我没有'找不到我想要的东西（或者可能不明白答案）。我猜问题出在 nd4j 后端或 jcublas 的问题上，但我不知道该怎么办。

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

【问题讨论】：

标签： java gpu deeplearning4j dl4j

【解决方案1】：

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

【讨论】：

我更改了 DeepLearning4j CUDA 特殊示例nd4j-cuda-10.2-platform int this DeepLearning4j CUDA 特例 nd4j-cuda-10.0-platform
在依赖项（在 pom 文件中）中，他们使用了 cuda 9.2、10.0、10.1 和 10.2，所以我删除了 9.2、10.1 和 10.2，但它仍然不起作用
这不是你在POM文件中指定依赖的方式，搜索“在Maven中指定依赖”
有效！它运行，我得到了混淆矩阵和所有......但我在初始化中收到这条消息“找不到cuDNN：通过包含deeplearning4j-cuda模块使用cuDNN以获得更好的GPU性能。有关更多信息，请参阅：@987654321 @" 我不明白，因为我按照他们在链接中所说的做了，但我仍然收到“找不到 cuDNN”消息
这只是一个提示。如果您安装了 cuDNN，您可以添加“deeplearning4j-cuda-10.x”依赖项，并提高 cuDNN 支持的选定操作的性能。