Update pre-requirements.txt

shenyunhang · Apr 12, 2024 · 8ea3c56 · 8ea3c56
1 parent 504a9ed
commit 8ea3c56
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/demo/pre-requirements.txt b/demo/pre-requirements.txt
@@ -1,4 +1,4 @@
 --index-url https://download.pytorch.org/whl/cu118
-torch==2.0.1
-torchvision==0.15.2
-torchaudio==2.0.2
+pytorch==2.2.1
+torchvision==0.17.1
+torchaudio==2.2.1