【发布时间】:2018-03-02 13:52:31
【问题描述】:
我正在尝试使用 GPU 运行使用 dl4j 构建的神经网络。使用本机平台依赖项时代码工作正常,但当我切换到使用 CUDA 时,执行失败并出现异常:
Caused by: java.lang.UnsatisfiedLinkError: ...\cuda-8.0-6.0-1.3-windows-x86_64.jar\org\bytedeco\javacpp\windows-x86_64\jnicuda.dll: Can't find dependent libraries
这是我的 build.gradle 中的内容:
// compile group: 'org.nd4j', name: 'nd4j-native-platform', version: '0.9.1'
compile group: 'org.nd4j', name: 'nd4j-cuda-8.0-platform', version: '0.9.1'
以下是 IDEA 中显示的依赖项:
我能够安装 CUDA Toolkit 并很好地构建示例。运行 deviceQuery 显示我的 CUDA 驱动程序/运行时版本是 9.1,这是一个问题,因为 nd4j-cuda-8.0-platform 引用 8.0?不确定,因为版本是 0.9.1,与我的 CUDA 版本匹配,而且似乎没有 9.0 平台可用。
还有什么我缺少的吗?如果是这样,我该如何解决/解决这个问题?
【问题讨论】:
-
尝试将 nvcc.exe 添加到 src/main/resources,如下所述:nd4j.org/getstarted.html#gpu 仍然是同样的错误。
-
嗯,基于 :nd4j.org/gpu_native_backends.html “我们目前支持 Cuda 7.5 和 8.0。”听起来 CUDA 9 不受支持且不向后兼容?
-
基于github.com/deeplearning4j/nd4j/issues/2285,主控支持CUDA 9。但是,编译组仍然存在相同的错误:'org.nd4j',名称:'nd4j-cuda-8.0-platform',版本:'0.9.2-SNAPSHOT'
-
CodeMonkeyCharlie 你能分享你的
gradle脚本吗?我面临同样的问题,但一直无法解决。另外,您提到您安装了 CUDA 8。您的意思是 CUDA Toolkit 8.0 GA2 从这里:developer.nvidia.com/cuda-80-ga2-download-archive
标签: cuda deeplearning4j dl4j