Skip to content

Commit 3e0bc07

Browse files
committed
update packages
1 parent e96859c commit 3e0bc07

File tree

1 file changed

+5
-16
lines changed

1 file changed

+5
-16
lines changed

vendor/ngc-pytorch/Dockerfile.21.10-py3

Lines changed: 5 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,8 @@ RUN apt-get update && \
5656
libsm6 \
5757
libtbb-dev
5858

59-
# Install CUDA-11.3 + cuDNN 8.2.0
60-
RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.2.0 /usr/local/cuda/lib64/libcudnn.so && \
59+
# Install CUDA-11.4 + cuDNN 8.2.0
60+
RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.2.4 /usr/local/cuda/lib64/libcudnn.so && \
6161
rm -rf /var/lib/apt/lists/* && \
6262
ldconfig
6363

@@ -118,24 +118,13 @@ RUN git clone --recursive https://github.com/bodono/scs-python.git && \
118118
RUN /opt/conda/bin/conda install -c conda-forge opencv ffmpeg spacy
119119

120120
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir \
121-
Cython==0.29.24 \
122-
tornado==6.1 \
123121
pystan==3.3tornado.0 \
124122
pycairo==1.20.1 \
125-
jupyter==1.0.0 \
126123
typeguard==2.12.1 \
127124
python-language-server[all] \
128125
pythran \
129126
matplotlib==3.4.3
130-
ENV SCIPY_VERSION 1.7.1
131-
# Install scipy
132-
RUN cd /tmp && \
133-
git clone --branch=v${SCIPY_VERSION} --depth=1 https://github.com/scipy/scipy.git scipy && \
134-
cd scipy && \
135-
git checkout -b v${SCIPY_VERSION} && \
136-
git submodule update --init && \
137-
cp site.cfg.example site.cfg && \
138-
python3 -m pip install -U --no-cache-dir \
127+
RUN python3 -m pip install -U --no-cache-dir \
139128
numpy==1.21.4 \
140129
pandas==1.3.4 \
141130
pythran \
@@ -146,7 +135,7 @@ RUN cd /tmp && \
146135
python3 setup.py install
147136

148137
WORKDIR /tmp
149-
COPY ./requirements.txt /tmp
138+
COPY ./requirements.21.10.txt /tmp/requirements.txt
150139
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir --ignore-installed -r requirements.txt && \
151140
rm -f /tmp/*.whl /tmp/requirements.txt
152141

@@ -235,7 +224,7 @@ LABEL ai.backend.kernelspec="1" \
235224
ai.backend.resource.min.cpu="1" \
236225
ai.backend.resource.min.mem="1g" \
237226
ai.backend.resource.min.cuda.device=1 \
238-
ai.backend.resource.min.cuda.shares=0.1 \
227+
ai.backend.resource.min.cuda.shares=0 \
239228
ai.backend.base-distro="ubuntu16.04" \
240229
ai.backend.runtime-type="python" \
241230
ai.backend.runtime-path="/opt/conda/bin/python3" \

0 commit comments

Comments
 (0)