From 3144e93edbfbc7f602493782d3446b67c0ceb33a Mon Sep 17 00:00:00 2001 From: Dirk Groeneveld Date: Wed, 3 Apr 2024 17:18:32 -0700 Subject: [PATCH] Update version of torch --- configs/mcli/mitchish7.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/mcli/mitchish7.yaml b/configs/mcli/mitchish7.yaml index 0b0592fbb..d359eb1e9 100644 --- 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