Update requirements.txt

yimutianyang · Aug 26, 2024 · aaefcec · aaefcec
1 parent fdd8605
commit aaefcec
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,8 @@ faiss==1.7.1
 numba==0.56.2
 scipy==1.3.0
 tqdm==4.62.3
+pytorch==1.10.0 
+torchvision==0.11.0 
+torchaudio==0.10.0 
+cudatoolkit=10.2