Added installation for torch

hotosm · Oct 22, 2024 · be8913a · be8913a
1 parent ac462ab
commit be8913a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -46,6 +46,9 @@ jobs:
       - name: Install tensorflow
         run: pip install tensorflow==2.9.2
 
+      - name : Install pytorch and ultralytics
+        run: pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 ultralytics==8.1.6
+
       - name: Install fair utilities
         run: pip install -e .