What are the highest versions of CUDA and cuDNN currently supported by DL4J?

DL4J
1

Does it support CUDA versions above 11.6, specifically 12.x?

1 Like
2

@cqiaoYc the next release will support 12. I"m currently in the middle of a rewrite so it might take a bit. I can show you the relevant bits if you want to build from source.

1 Like
3

Thank you! I’m currently using CUDNN 11.8 on Ubuntu 2024.04, and it’s running smoothly.

4

I was looking to play with DL4J / ND4J using CUDA, but I have version 12…

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Fri_Jan__6_16:45:21_PST_2023
Cuda compilation tools, release 12.0, V12.0.140
Build cuda_12.0.r12.0/compiler.32267302_0

@agibsonccc I’m not sure I’d be in a position to help move this ticket along, but maybe in the least provide another environment to test.

5

I’ve had it building with cuda 12.x on my development branch for a while. I’d be happy to help you build from source or support a standalone pull request if you’d like to try it.

6

Hi @agibsonccc, Yes that would be great. Just let me know which way works best for you.

7

@sradford I’ve recently put up a series of PRs split off from my main development branch that will allow cuda 12.1 and 12.3. Let me get that merged to master in the next day or 2 then I’ll help you with instructions.

1 Like
8

@sradford here’s a branch you can try. I’ll be merging this with @treo gets back to me.