From 90d2c04bb5a1f731991ce9b6965d60f6d033bf4e Mon Sep 17 00:00:00 2001
From: Philmod <philmod@google.com>
Date: Tue, 18 Oct 2022 10:12:53 -0400
Subject: [PATCH] Pin accelerate which breaks catalyst on 2xT4 GPUs

---
 Dockerfile.tmpl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 9c6800fd..6eab7ca0 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -482,6 +482,8 @@ RUN pip install flashtext && \
     pip install optuna && \
     pip install plotly_express && \
     pip install albumentations && \
+    # b/254245259 catalyst requires accelerate but it breaks with the version 0.13.1
+    pip install accelerate==0.12.0 && \
     pip install catalyst && \
     # b/206990323 osmx 1.1.2 requires numpy >= 1.21 which we don't want. 
     pip install osmnx==1.1.1 && \