Skip to content

Fix spacy by pinning mlk version #1126

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 15 commits into from
Jan 25, 2022
Merged

Fix spacy by pinning mlk version #1126

Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
18b8c82
Post jenkins updates to Mattermost
Philmod Jan 19, 2021
9f07943
merge
Philmod Jan 21, 2021
529e2f8
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Jan 21, 2021
9fafd22
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Jan 26, 2021
cbc44e5
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Jan 26, 2021
f11ebe8
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Mar 16, 2021
aefe69e
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Mar 16, 2021
0fecbf8
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Mar 22, 2021
c7d489e
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod Apr 21, 2021
1ac71a6
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod May 13, 2021
31a1e32
Merge branch 'master' of github.com:Kaggle/docker-python
Philmod May 26, 2021
8407a47
Merge branch 'main' of github.com:Kaggle/docker-python
Philmod Oct 12, 2021
4758ca4
Merge branch 'main' of github.com:Kaggle/docker-python
Philmod Dec 6, 2021
6bfd8e0
Merge branch 'main' of github.com:Kaggle/docker-python
Philmod Jan 25, 2022
bf6fdac
Pin mkl version and properly install Spacy for GPU
Philmod Jan 25, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 11 additions & 1 deletion Dockerfile.tmpl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,8 @@ RUN conda config --add channels nvidia && \
conda config --add channels rapidsai && \
# Base image channel order: conda-forge (highest priority), defaults.
# End state: rapidsai (highest priority), nvidia, conda-forge, defaults.
conda install mkl cartopy=0.19 imagemagick=7.1 pyproj==3.1.0 && \
# b/216162758 Pin mkl which last version breaks spacy.
conda install mkl=2021.4.0 cartopy=0.19 imagemagick=7.1 pyproj==3.1.0 && \
/tmp/clean-layer.sh

{{ if eq .Accelerator "gpu" }}
Expand Down Expand Up @@ -130,6 +131,15 @@ RUN pip install mxnet && \
/tmp/clean-layer.sh
{{ end}}

# Install spacy
{{ if eq .Accelerator "gpu" }}
RUN pip install spacy[cuda$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION] && \
/tmp/clean-layer.sh
{{ else }}
RUN pip install spacy && \
/tmp/clean-layer.sh
{{ end}}

# Install GPU specific packages
{{ if eq .Accelerator "gpu" }}
# Install GPU-only packages
Expand Down