Skip to content

Commit fe78d6a

Browse files
committed
NGC pytorch 21.09
1 parent 34a6cc8 commit fe78d6a

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

vendor/ngc-pytorch/Dockerfile.21.09-py3

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ RUN apt-get update && \
2020
fonts-nanum-extra \
2121
gfortran \
2222
htop \
23+
ffmpeg \
2324
libasound2-dev \
2425
libatlas-base-dev \
2526
libavresample-dev \
@@ -113,16 +114,17 @@ RUN git clone --recursive https://github.com/bodono/scs-python.git && \
113114
cd /tmp/scs-python && \
114115
python setup.py install --scs --gpu
115116

116-
RUN /opt/conda/bin/conda install -c conda-forge opencv ffmpeg
117+
RUN /opt/conda/bin/conda install -c conda-forge opencv ffmpeg spacy
117118

118119
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir \
119120
Cython==0.29.24 \
120121
tornado==6.1 \
121122
pystan==3.2.0 \
122-
pycairo==1.20.1typeguard \
123+
pycairo==1.20.1 \
123124
jupyter==1.0.0 \
124125
typeguard==2.12.1 \
125-
python-language-server[all] \
126+
python-language-server[all] \
127+
pythran \
126128
matplotlib==3.4.3
127129
ENV SCIPY_VERSION 1.7.1
128130
# Install scipy
@@ -134,7 +136,7 @@ RUN cd /tmp && \
134136
python3 -m pip install -U --no-cache-dir \
135137
numpy==1.21.2 \
136138
pandas==1.3.3 \
137-
scikit-learn==ᇂ1.0 \
139+
scikit-learn==1.0 \
138140
hypothesis==6.23.0 \
139141
python-lsp-server \
140142
&& \
@@ -215,7 +217,7 @@ RUN apt autoclean && \
215217

216218
RUN /opt/conda/bin/python3 -m ipykernel install \
217219
--prefix=/opt/conda/ \
218-
--display-name "PyTorch 1.10 (NGC 21.09/Python 3.8 Conda) on Backend.AI" && \
220+
--display-name "PyTorch 1.10 (NGC 21.07/Python 3.8 Conda) on Backend.AI" && \
219221
cat /opt/conda/share/jupyter/kernels/python3/kernel.json
220222

221223
# Backend.AI specifics
@@ -235,6 +237,8 @@ LABEL ai.backend.kernelspec="1" \
235237
ai.backend.runtime-type="python" \
236238
ai.backend.runtime-path="/opt/conda/bin/python3" \
237239
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8091,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:preopen:5000,nniboard:preopen:8080"
238-
240+
241+
RUN python3 -m spacy download en_core_web_sm && \
242+
python -m spacy project clone pipelines/tagger_parser_predicted_annotations
239243
WORKDIR /home/work
240244
# vim: ft=dockerfile

0 commit comments

Comments
 (0)