Python 3.12 support (#657)

PyTorch 2.2.2 TensorFlow 2.16.2 CUDA 12.1
isl-org · Oct 8, 2024 · 3ad71dd · 3ad71dd
1 parent 98f0781
commit 3ad71dd
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/requirements-tensorflow.txt b/requirements-tensorflow.txt
@@ -1 +1 @@
-tensorflow~=2.13.0
+tensorflow~=2.16.2
diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
@@ -1,5 +1,5 @@
 -f https://download.pytorch.org/whl/torch/
-torch==2.0.1+cu117
+torch==2.2.2+cu121
 -f https://download.pytorch.org/whl/torchvision/
-torchvision==0.15.2+cu117
+torchvision==0.17.2+cu121
 tensorboard
diff --git a/requirements-torch.txt b/requirements-torch.txt
@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu/
-torch==2.0.1+cpu ; sys_platform != 'darwin'
-torchvision==0.15.2+cpu ; sys_platform != 'darwin'
-torch==2.0.1 ; sys_platform == 'darwin'
-torchvision==0.15.2 ; sys_platform == 'darwin'
+torch==2.2.2+cpu ; sys_platform != 'darwin'
+torchvision==0.17.2+cpu ; sys_platform != 'darwin'
+torch==2.2.2 ; sys_platform == 'darwin'
+torchvision==0.17.2 ; sys_platform == 'darwin'
 tensorboard