Update version of torch

allenai · Apr 4, 2024 · 3144e93 · 3144e93
1 parent d49a91d
commit 3144e93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/mcli/mitchish7.yaml b/configs/mcli/mitchish7.yaml
@@ -1,5 +1,5 @@
 name: olmo-7b
-image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04
+image: mosaicml/pytorch:2.2.1_cu121-python3.11-ubuntu20.04
 compute:
   cluster: r15z4
   gpus: 512