From c3082ab373f6d175c464b6a461140741f77bbc01 Mon Sep 17 00:00:00 2001
From: Maxwell Lou <maxlou2005@gmail.com>
Date: Mon, 19 Aug 2024 22:06:20 -0400
Subject: [PATCH] update to pytorch 2,4.0 for python 3.11 support

---
 .github/workflows/run-tests.yml | 6 +++---
 pyproject.toml                  | 4 ++--
 requirements.txt                | 6 +++---
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
index 7c598af..b97f2a7 100644
--- a/.github/workflows/run-tests.yml
+++ b/.github/workflows/run-tests.yml
@@ -1,4 +1,4 @@
-# This workflow will install Python dependencies and run tests with PyTest using Python 3.8
+# This workflow will install Python dependencies and run tests with PyTest using Python 3.11
 # For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
 
 name: Run tests
@@ -18,10 +18,10 @@ jobs:
       - uses: actions/checkout@v3
 
       # Set Python version
-      - name: Set up Python 3.8
+      - name: Set up Python 3.11
         uses: actions/setup-python@v4
         with:
-          python-version: 3.8
+          python-version: 3.11
 
       # Set up submodules and submodule dependencies
       # TODO: Uncomment when there are submodules
diff --git a/pyproject.toml b/pyproject.toml
index 4927a25..e081f06 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,7 +28,7 @@ jobs = 0
 
 # Minimum Python version to use for version dependent checks. Will default to the
 # version used to run pylint.
-py-version = "3.8"
+py-version = "3.11"
 
 # Discover python modules and packages in the file system subtree.
 recursive = true
@@ -105,7 +105,7 @@ addopts = "--ignore=modules/common/"
 
 [tool.black]
 line-length = 100
-target-version = ["py38"]
+target-version = ["py311"]
 # Excludes files or directories in addition to the defaults
 # Submodules
 extend-exclude = "modules/common/*"
diff --git a/requirements.txt b/requirements.txt
index b3a2f48..e8c203e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,9 +1,9 @@
 # Packages listed in alphabetical order
 
 # For any non-Jetson computer (i.e. all developers)
---extra-index-url https://download.pytorch.org/whl/cu117
-torch==1.13.1+cu117
-torchvision==0.14.1+cu117
+--extra-index-url https://download.pytorch.org/whl/cu124
+torch==2.4.0
+torchvision==0.19.0
 ultralytics
 
 # Linters and formatters are explicitly versioned