Update pytorch image to torch version from 2.0 to 2.2 on 2023b release

opendatahub-io · Jun 18, 2024 · a3cb0bc · a3cb0bc
1 parent d9e264b
commit a3cb0bc
Show file tree

Hide file tree

Showing 6 changed files with 360 additions and 301 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"