From 9214b6543fa652215e690c94d1b2ef370e3209cc Mon Sep 17 00:00:00 2001
From: Kshitij Raj Sharma <36752999+kshitijrajsharma@users.noreply.github.com>
Date: Wed, 30 Oct 2024 20:07:07 +0100
Subject: [PATCH] Update build.yml

---
 .github/workflows/build.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 .