Apply opendatahub-io#656 to Python 3.11 images

caponetto · Aug 6, 2024 · c465b79 · c465b79
1 parent 1a12b8b
commit c465b79
Show file tree

Hide file tree

Showing 5 changed files with 497 additions and 513 deletions.
diff --git a/jupyter/rocm/pytorch/ubi9-python-3.11/Pipfile b/jupyter/rocm/pytorch/ubi9-python-3.11/Pipfile
@@ -4,7 +4,8 @@ verify_ssl = true
 name = "pypi"
 
 [[source]]
-url = "https://download.pytorch.org/whl/rocm6.0/"
+# Please keep in sync with rocm/ubi9-python-3.11/Dockerfile
+url = "https://download.pytorch.org/whl/rocm6.1/"
 verify_ssl = true
 name = "pytorch"
 
@@ -13,9 +14,9 @@ name = "pytorch"
 [packages]
 # PyTorch packages
 tensorboard = "~=2.16.2"
-torch = {version = "~=2.3.1", index = "pytorch"}
-pytorch-triton-rocm = {version = "~=2.3.1", index = "pytorch"}
-torchvision = {version = "~=0.18.1", index = "pytorch"} # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
+torch = {version = "~=2.4.0", index = "pytorch"}
+torchvision = {version = "~=0.19.0", index = "pytorch"} # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
+pytorch-triton-rocm = {version = "~=3.0.0", index = "pytorch"}
 # Datascience and useful extensions
 boto3 = "~=1.34.50"
 kafka-python = "~=2.0.2"