diff --git a/pyproject.toml b/pyproject.toml index b293db8..c9104e1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -26,7 +26,7 @@ dependencies = [ ] [project.optional-dependencies] -cuda = ["onnx", "onnxruntime-cuda"] +cuda = ["onnx", "onnxruntime-gpu"] directml = ["onnx", "onnxruntime-directml"] openvino = ["onnx", "onnxruntime-openvino"]