Aidge_backend_cuda - unable to set model on cuda

#Aidge tutorial - Learning

Environnement

• aidge 0.4.0

• aidge_backend_cpu 0.4.0

• aidge_backend_cuda 0.4.0

• aidge_core 0.4.0

• aidge_learning 0.2.2

• aidge_onnx 0.4.0

• CUDA version : 12.7

• cuDNN 9.6.0 with CUDA 12 specific package

Put a model on cuda

The goal of the tutorial is to use the backend cuda. I have an error in the tutorial when I want to chose to set the backend for my model : Cell :

model = aidge_onnx.load_onnx("lenet.onnx")
aidge_core.remove_flatten(model)

# Configure the model
model.set_datatype(aidge_core.dtype.float32)
model.set_backend("cuda")

Error:

Assertion failed: exists(key) in aidge/aidge_core/include/aidge/utils/Registrar.hpp:79
missing or invalid registrar key: ("cuda", "Float32") for registrable object N5Aidge6TensorE
Did you include/import the corresponding module?
If so, it is possible that the object is not yet supported.

Problem

I cannot use my model on GPU. Where does this error come from? An environnement incompatibility or something else?