Pin to use 2.5

- Update torch to 2.5 - Update torchrec to use 1.0.0 - Update fbgemm to use 1.0.0
pytorch · Oct 14, 2024 · dc4ea7a · dc4ea7a
1 parent ecfbe26
commit dc4ea7a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
@@ -14,7 +14,7 @@ tqdm==4.66.1
 numpy==1.24.4
 matplotlib
 librosa
-torch==2.4
+torch==2.5
 torchvision
 torchtext
 torchdata
@@ -70,5 +70,5 @@ pycocotools
 semilearn==0.3.2
 torchao==0.5.0
 segment_anything==1.0
-torchrec==0.8.0
-fbgemm-gpu==0.8.0
+torchrec==1.0.0
+fbgemm-gpu==1.0.0