requirements update

Aske-Rosted · May 16, 2024 · 0750dc6 · 0750dc6
1 parent c579dc6
commit 0750dc6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/requirements/torch_cu118.txt b/requirements/torch_cu118.txt
@@ -1,4 +1,5 @@
 # Contains packages requirements for GPU installation
 --find-links https://download.pytorch.org/whl/torch_stable.html
 torch==2.2.0+cu118
+torchvision==0.17.0+cu118
 --find-links https://data.pyg.org/whl/torch-2.2.0+cu118.html
diff --git a/requirements/torch_cu121.txt b/requirements/torch_cu121.txt
@@ -1,4 +1,5 @@
 # Contains packages requirements for GPU installation
 --find-links https://download.pytorch.org/whl/torch_stable.html
 torch==2.2.0+cu121
+torchvision==0.17.0+cu121
 --find-links https://data.pyg.org/whl/torch-2.2.0+cu121.html