fix(cuda12): pin version (#852)

rungalileo · Apr 24, 2024 · e2b1da3 · e2b1da3
1 parent 61e1b98
commit e2b1da3
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 18 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -52,13 +52,13 @@ setfit = {version ="==0.7.0", optional = true}
 sentencepiece = "^0.2.0"
 
 # cuda dependencies
-ucx-py-cu12 = { version ="^0.36", source = "nvidia", optional = true }
-rmm-cu12 = {version = "^24.2.0", source = "nvidia", optional=true}
-raft-dask-cu12 = {version = "^24.2.0", source = "nvidia", optional=true}
-pylibraft-cu12 = {version = "^24.2.0", source = "nvidia", optional=true}
-dask-cudf-cu12 = {version = "^24.2.0", source = "nvidia", optional=true}
-cudf-cu12 = {version = "^24.2.0", source = "nvidia", optional=true}
-cuml-cu12 = {version = "^24.2.0", source = "nvidia", optional=true}
+ucx-py-cu12 = { version ="==0.36", source = "nvidia", optional = true }
+rmm-cu12 = {version = "==24.2.0", source = "nvidia", optional=true}
+raft-dask-cu12 = {version = "==24.2.0", source = "nvidia", optional=true}
+pylibraft-cu12 = {version = "==24.2.0", source = "nvidia", optional=true}
+dask-cudf-cu12 = {version = "==24.2.0", source = "nvidia", optional=true}
+cudf-cu12 = {version = "==24.2.0", source = "nvidia", optional=true}
+cuml-cu12 = {version = "==24.2.0", source = "nvidia", optional=true}
 tensorflow = ">=2.9.1,<2.15.0"