Update the ROCm versions installed from 6.0 to 6.1 to make it in sync…

… with what we use in Dockerfile (#656)

* Sync the ROCm versions used among our Pipfiles and Dockerfiles to 6.1

* Add a comments to point to the points of sync for ROCM

jupyter/rocm/pytorch/ubi9-python-3.9/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.9/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"}
-torchvision = {version = "~=0.18.1", index = "pytorch"} # https://pytorch.org/get-started/previous-versions/#linux-and-windows-1
-pytorch-triton-rocm = {version = "~=2.3.1", index = "pytorch"}
+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"