Create torch.txt

WildMeOrg · Jun 15, 2024 · 8904210 · 8904210
1 parent acc2726
commit 8904210
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/requirements/torch.txt b/requirements/torch.txt
@@ -0,0 +1,4 @@
+--index-url https://download.pytorch.org/whl/cu121
+torch==2.1.0 
+torchvision==0.16.0
+torchaudio==2.1.0