From 3ad71dd62d67302b0ea75bf0fde560dc8259c692 Mon Sep 17 00:00:00 2001
From: Sameer Sheorey <41028320+ssheorey@users.noreply.github.com>
Date: Tue, 8 Oct 2024 08:40:54 -0700
Subject: [PATCH] Python 3.12 support (#657)

PyTorch 2.2.2
TensorFlow 2.16.2
CUDA 12.1
---
 requirements-tensorflow.txt | 2 +-
 requirements-torch-cuda.txt | 4 ++--
 requirements-torch.txt      | 8 ++++----
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/requirements-tensorflow.txt b/requirements-tensorflow.txt
index 742e3f30..82f90bb1 100644
--- 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
index b69670fc..bdef0da6 100644
--- 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
index fb417a31..bc863461 100644
--- 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