diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 19792e5..dde280a 100644
--- 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.17.0 torchaudio==2.2.0 --extra-index-url https://download.pytorch.org/whl/cu113 ultralytics==8.1.6
+        run: pip install torch==2.4.0 torchvision==0.18.0 torchaudio==2.2.0 --extra-index-url https://download.pytorch.org/whl/cu113 ultralytics==8.1.6
 
       - name: Install fair utilities
         run: pip install -e .