From b6a2ecf101b557ddc15e7fb266b63dfadd0a1fb9 Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Mon, 24 Aug 2020 17:28:05 +0000 Subject: [PATCH 1/2] Pin kornia to prevent torch upgrade. BUG=162850432 --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 245a2edc..3bb9fa4a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -343,7 +343,8 @@ RUN pip install bcolz && \ pip install nbformat && \ pip install notebook==5.5.0 && \ pip install olefile && \ - pip install kornia && \ + # b/162850432 unpin once we upgrade to pytorch 1.6 + pip install kornia==0.3.2 && \ pip install pandas_summary && \ pip install pandocfilters && \ pip install pexpect && \ From 187af5a659ccf374e48473b0a63845f30f840222 Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Mon, 24 Aug 2020 20:13:12 +0000 Subject: [PATCH 2/2] Pin fastai --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 3bb9fa4a..fe92f1d1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -374,7 +374,8 @@ RUN pip install bcolz && \ pip install widgetsnbextension && \ pip install pyarrow && \ pip install feather-format && \ - pip install fastai && \ + # b/162850432 unpin once we upgrade to pytorch 1.6 + pip install fastai~=1.0.61 && \ pip install allennlp && \ python -m spacy download en && python -m spacy download en_core_web_lg && \ apt-get install -y ffmpeg && \