From a295b0c29d2951c4e5f1a0bb572fd060fc12a5a7 Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <38561704+bertsky@users.noreply.github.com>
Date: Wed, 9 Oct 2024 16:32:57 +0200
Subject: [PATCH] deps-torch: also install torchvision

---
 Makefile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index b5cd2f276..26524a9fa 100644
--- a/Makefile
+++ b/Makefile
@@ -158,7 +158,7 @@ deps-tf2:
 	fi
 
 deps-torch:
-	$(PIP) install -i https://download.pytorch.org/whl/cu118 torch
+	$(PIP) install -i https://download.pytorch.org/whl/cu118 torch torchvision
 
 # Dependencies for deployment in an ubuntu/debian linux
 deps-ubuntu: