@@ -5,6 +5,7 @@ FROM nvcr.io/nvidia/pytorch:21.03-py3
5
5
6
6
ENV DEBIAN_FRONTEND=noninteractive \
7
7
MPLBACKEND=Svg \
8
+ PIP_IGNORE_INSTALLED=0 \
8
9
PYTHONUNBUFFERED=1 \
9
10
LD_LIBRARY_PATH="/usr/local/cuda/compat/lib:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
10
11
PATH="/usr/local/nvm/versions/node/v14.8.0/bin:/opt/conda/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin:/usr/local/src/lightgbm/LightGBM:/usr/local/bin/mecab" \
@@ -155,7 +156,6 @@ RUN cd /tmp && \
155
156
WORKDIR /tmp
156
157
COPY ./requirements.txt /tmp
157
158
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir --ignore-installed -r requirements.txt && \
158
- /opt/conda/bin/python3 -m pip install --no-cache-dir tensorflow_model_analysis && \
159
159
rm -f /tmp/*.whl /tmp/requirements.txt
160
160
161
161
# install git-lfs
@@ -167,7 +167,7 @@ RUN curl -sLO https://github.com/git-lfs/git-lfs/releases/download/v2.13.3/git-l
167
167
168
168
COPY ./service-defs /etc/backend.ai/service-defs
169
169
RUN curl -fL https://github.com/cdr/code-server/releases/download/v3.9.3/code-server-3.9.3-linux-amd64.tar.gz | tar -C /usr/local/lib -xz && \
170
- mv /usr/local/lib/code-server-3.9.3-linux-amd64 /usr/local/lib/code-server-3.9.2 && \
170
+ mv /usr/local/lib/code-server-3.9.3-linux-amd64 /usr/local/lib/code-server-3.9.3 && \
171
171
ln -s /usr/local/lib/code-server-3.9.3/bin/code-server /usr/local/bin/code-server
172
172
# Install Open MPI
173
173
RUN mkdir /tmp/openmpi && \
@@ -214,19 +214,17 @@ RUN jupyter nbextensions_configurator enable && \
214
214
jupyter nbextension enable toc2/main && \
215
215
jupyter labextension install @pyviz/jupyterlab_pyviz && \
216
216
jupyter labextension install @bokeh/jupyter_bokeh && \
217
- jupyter labextension install --no-build jupyterlab-nvdashboard && \
218
217
jupyter lab build
219
218
220
219
RUN apt autoclean && \
221
220
sed -i 's/source \/ usr\/ local\/ nvm\/ nvm.sh//' /etc/bash.bashrc && \
222
- python3 -m pip uninstall -y tensorboard-plugin-wit && \
223
221
rm -rf /var/lib/apt/lists/* && \
224
222
rm -rf /root/.cache && \
225
223
rm -rf /tmp/*
226
224
227
225
RUN /opt/conda/bin/python3 -m ipykernel install \
228
226
--prefix=/opt/conda/ \
229
- --display-name "PyTorch 1.8 (NGC 21.03/Python 3.8 Conda) on Backend.AI" && \
227
+ --display-name "PyTorch 1.9 (NGC 21.03/Python 3.8 Conda) on Backend.AI" && \
230
228
cat /opt/conda/share/jupyter/kernels/python3/kernel.json
231
229
232
230
# Backend.AI specifics
0 commit comments