feat: sync pytorch for docker with version used in prepare_environment

2026-08-16 13:13:16 +02:00 · 2024-05-22 23:50:27 +02:00
parent 299ba4d2a2
commit 7f6dfe25a5
2 changed files with 3 additions and 6 deletions
@@ -1,5 +1,2 @@
-torch==2.0.1
-torchvision==0.15.2
-torchaudio==2.0.2
-torchtext==0.15.2
-torchdata==0.6.1
+torch==2.1.0
+torchvision==0.16.0