Skip to content

Commit 3373727

Browse files
committed
RHAIENG-1504: copy the rhds-only ROCm update for torch back into odh-io
* red-hat-data-services#1615
1 parent 9bfef1f commit 3373727

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

jupyter/rocm/pytorch/ubi9-python-3.12/pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ requires-python = "==3.12.*"
66
dependencies = [
77
# ROCm PyTorch packages
88
"tensorboard~=2.20.0",
9-
"torch==2.7.1+rocm6.2.4",
10-
"torchvision==0.22.1+rocm6.2.4", # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
9+
"torch==2.7.1+rocm6.3",
10+
"torchvision==0.22.1+rocm6.3", # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
1111
"pytorch-triton-rocm~=3.3.1",
1212

1313
# Datascience and useful extensions
@@ -61,7 +61,7 @@ pytorch-triton-rocm = { index = "pytorch-rocm" }
6161
# Please keep in sync with rocm/ubi9-python-3.12/Dockerfile
6262
[[tool.uv.index]]
6363
name = "pytorch-rocm"
64-
url = "https://download.pytorch.org/whl/rocm6.2.4"
64+
url = "https://download.pytorch.org/whl/rocm6.3"
6565
explicit = true
6666

6767
[tool.uv]

runtimes/rocm-pytorch/ubi9-python-3.12/pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ requires-python = "==3.12.*"
66
dependencies = [
77
# ROCm PyTorch packages
88
"tensorboard~=2.20.0",
9-
"torch==2.7.1+rocm6.2.4",
10-
"torchvision==0.22.1+rocm6.2.4", # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
9+
"torch==2.7.1+rocm6.3",
10+
"torchvision==0.22.1+rocm6.3", # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
1111
"pytorch-triton-rocm~=3.3.1",
1212

1313
# Datascience and useful extensions
@@ -63,7 +63,7 @@ pytorch-triton-rocm = { index = "pytorch-rocm" }
6363
# Please keep in sync with rocm/ubi9-python-3.12/Dockerfile
6464
[[tool.uv.index]]
6565
name = "pytorch-rocm"
66-
url = "https://download.pytorch.org/whl/rocm6.2.4"
66+
url = "https://download.pytorch.org/whl/rocm6.3"
6767
explicit = true
6868

6969
[tool.uv]

0 commit comments

Comments
 (0)