Skip to content

Flux on GPU is incompatible with NVIDIA driver version 13 #2618

@jacob-m-wilson-42

Description

@jacob-m-wilson-42

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 cpu

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions