java - 尝试使用 Cuda 运行 Nd4j 时出错

Question

我已经遵循了我能找到的所有教程，但不知怎的，我似乎无法让 Nd4j 与 Cuda 后端一起工作。我想我缺少一个库，但我不知道是哪个库或在哪里下载它。

我收到的错误是:

18:23:23.872 [main] WARN  org.nd4j.linalg.factory.Nd4jBackend - Skipped [JCublasBackend] backend (unavailable): java.lang.UnsatisfiedLinkError: no cudart in java.library.path
18:23:23.958 [main] INFO  org.nd4j.linalg.factory.Nd4jBackend - Loaded [CpuBackend] backend

我正在使用 Nd4j，并且我的类路径上有(以及一大堆其他)以下 jar:

nd4j-native-1.0.0-beta3.jar
nd4j-native-api-1.0.0-beta3.jar
cuda-10.0-7.3-1.4.3.jar
nd4j-cuda-10.0-1.0.0-beta3.jar. 

Cuda 10.0安装成功:

C:\Users\username>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:04_Central_Daylight_Time_2018
Cuda compilation tools, release 10.0, V10.0.130

我尝试将 Java 指向 Cuda bin 目录，如下所示:

-Djava.library.path="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin"

但这没有什么区别。此外，我在 nd4j-cuda-10.0-platform-1.0.0-beta3.jar 和 cuda-platform-10.0-7.3-1.4.3.jar 内部的 pom.xml 上尝试了 mvn 命令，这导致下载很多 jars，但它们似乎都不包含我缺少的 dll。

在搜索时，我发现其他用户正在谈论像“cuda-10.0-7.3-1.4.3-windows-x86_64.jar”这样的 jar，它应该包含 jnicudart.dll，但我在任何地方都找不到这样的 jar。我在这里缺少什么？

--更新--我已将所有内容更新到 beta5，并发现一个包含缺少的 jnicudart.dll 的 cuda jar，但我仍然遇到相同的错误。我使用依赖项来检查问题所在，结果如下:

它说 NVidia cudart64_100.dll 的校验和不正确。这是否与 NVidia Cuda10 不支持我的 Geforce GT 730 有关，因为它是 Fermi 架构？或者cuda安装是否损坏或者什么的？

Answer 1

1. 请将您的 nd4j/dl4j 版本更新至 1.0.0-beta5
2. 请告诉我，您的 GPU 型号名称和 nvidia 驱动程序版本是什么？即显示 nvidia-smi 输出。