Fix torch version

hotosm · Oct 30, 2024 · 57bbf69 · 57bbf69
1 parent fd89269
commit 57bbf69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -47,7 +47,7 @@ jobs:
         run: pip install tensorflow==2.9.2
 
       - name : Install pytorch and ultralytics
-        run: pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.3.0 --extra-index-url https://download.pytorch.org/whl/cu113 ultralytics==8.1.6
+        run: pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --extra-index-url https://download.pytorch.org/whl/cu113 ultralytics==8.1.6
 
       - name: Install fair utilities
         run: pip install -e .