Skip to content

Commit 1517713

Browse files
authored
Pin accelerate which breaks catalyst on 2xT4 GPUs (#1188)
1 parent d7d0794 commit 1517713

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Dockerfile.tmpl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -482,6 +482,8 @@ RUN pip install flashtext && \
482482
pip install optuna && \
483483
pip install plotly_express && \
484484
pip install albumentations && \
485+
# b/254245259 catalyst requires accelerate but it breaks with the version 0.13.1
486+
pip install accelerate==0.12.0 && \
485487
pip install catalyst && \
486488
# b/206990323 osmx 1.1.2 requires numpy >= 1.21 which we don't want.
487489
pip install osmnx==1.1.1 && \

0 commit comments

Comments
 (0)