Merge pull request #572 from atheo89/RHOAIENG-8469

Update pytorch image to torch version from 2.0 to 2.2 on 2023b release
opendatahub-io · Jun 18, 2024 · 9fe149c · 9fe149c
2 parents e521c80 + 15ab52a
commit 9fe149c
Show file tree

Hide file tree

Showing 6 changed files with 359 additions and 299 deletions.
diff --git a/jupyter/pytorch/ubi9-python-3.9/Pipfile b/jupyter/pytorch/ubi9-python-3.9/Pipfile
@@ -13,8 +13,8 @@ name = "pytorch"
 [packages]
 # PyTorch packages
 tensorboard = "~=2.13.0"
-torch = {version = "~=2.0.0", index = "pytorch"}
-torchvision = {version = "~=0.15.1", index = "pytorch"}
+torch = {version = "~=2.2.2", index = "pytorch"}
+torchvision = {version = "~=0.17.2", index = "pytorch"}
 # Datascience and useful extensions
 boto3 = "~=1.28.41"
 kafka-python = "~=2.0.2"