[ci/requirements] Upgrade torch to 2.0.1

python/requirements/ml/dl-cpu-requirements.txt

@@ -11,12 +11,12 @@ tensorflow-datasets
 
 --extra-index-url https://download.pytorch.org/whl/cpu # for CPU versions of torch, torchvision
 --find-links https://data.pyg.org/whl/torch-1.13.0+cpu.html # for CPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv
-torch==1.13.0
-torchmetrics==0.9.3
-torchtext==0.14.0
-torchvision==0.14.0
-torch-scatter==2.1.0
-torch-sparse==0.6.16
-torch-cluster==1.6.0
-torch-spline-conv==1.2.1
-torch-geometric==2.1.0
+torch==2.0.1
+torchmetrics==0.11.4
+torchtext==0.15.2
+torchvision==0.15.2
+torch-scatter==2.1.1
+torch-sparse==0.6.17
+torch-cluster==1.6.1
+torch-spline-conv==1.2.2
+torch-geometric==2.3.1

python/requirements/ml/dl-gpu-requirements.txt

@@ -1,11 +1,11 @@
 # If you make changes below this line, please also make the corresponding changes to `dl-cpu-requirements.txt`!
 
---extra-index-url https://download.pytorch.org/whl/cu116 # for GPU versions of torch, torchvision
---find-links https://data.pyg.org/whl/torch-1.13.0+cu116.html # for GPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv
+--extra-index-url https://download.pytorch.org/whl/cu118 # for GPU versions of torch, torchvision
+--find-links https://data.pyg.org/whl/torch-2.0.1+cu118.html # for GPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv
 # specifying explicit plus-notation below so pip overwrites the existing cpu verisons
-torch==1.13.0+cu116
-torchvision==0.14.0+cu116
-torch-scatter==2.1.0+pt113cu116
-torch-sparse==0.6.15+pt113cu116
-torch-cluster==1.6.0+pt113cu116
-torch-spline-conv==1.2.1+pt113cu116
+torch==2.0.1+cu118
+torchvision==0.15.2+cu118
+torch-scatter==2.1.1+pt113cu118
+torch-sparse==0.6.17+pt113cu118
+torch-cluster==1.6.1+pt113cu118
+torch-spline-conv==1.2.2+pt113cu118