feat(machine-learning): support cuda 12 (#7569)

* feat: support cuda12

* fix: group optional

* move to cuda 12

* pr feedback
This commit is contained in:
martin
2024-03-03 05:36:16 +01:00
committed by GitHub
parent fd3503e77d
commit 8ce18b3403
3 changed files with 25 additions and 16 deletions
+6 -1
View File
@@ -45,7 +45,7 @@ onnxruntime = "^1.15.0"
optional = true
[tool.poetry.group.cuda.dependencies]
onnxruntime-gpu = "^1.15.0"
onnxruntime-gpu = {version = "^1.17.0", source = "cuda12"}
[tool.poetry.group.openvino]
optional = true
@@ -59,6 +59,11 @@ optional = true
[tool.poetry.group.armnn.dependencies]
onnxruntime = "^1.15.0"
[[tool.poetry.source]]
name = "cuda12"
url = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"
priority = "explicit"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"