-
-
Couldn't load subscription status.
- Fork 615
Open
Description
The issue lies with cuDNN. cuDNN only supports up to CUDA version 5.8.2, however, CUDA version 5.8.3 is the version required to support CUDA driver version 13. Apologies if this is already being worked on, I thought I would just bring this to attention. Please see the below MWE that replicates the issue.
using Pkg
Pkg.activate("test_env")
Pkg.add("CUDA") # Version 5.6.3 required for CUDA drivers version 13
Pkg.add("Flux")
using CUDA
using Flux # throws a warning that cuDNN is not installed, however, cuDNN is not compatible with CUDA v5.6.3
test = Dense(3=>1) # create a simple dense layer to test
test = gpu(test) # attempt to move model onto GPU, throws warning stating the model defaulted to cpuMetadata
Metadata
Assignees
Labels
No labels