@@ -56,8 +56,8 @@ RUN apt-get update && \
56
56
libsm6 \
57
57
libtbb-dev
58
58
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 && \
61
61
rm -rf /var/lib/apt/lists/* && \
62
62
ldconfig
63
63
@@ -118,24 +118,13 @@ RUN git clone --recursive https://github.com/bodono/scs-python.git && \
118
118
RUN /opt/conda/bin/conda install -c conda-forge opencv ffmpeg spacy
119
119
120
120
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir \
121
- Cython==0.29.24 \
122
- tornado==6.1 \
123
121
pystan==3.3tornado.0 \
124
122
pycairo==1.20.1 \
125
- jupyter==1.0.0 \
126
123
typeguard==2.12.1 \
127
124
python-language-server[all] \
128
125
pythran \
129
126
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 \
139
128
numpy==1.21.4 \
140
129
pandas==1.3.4 \
141
130
pythran \
@@ -146,7 +135,7 @@ RUN cd /tmp && \
146
135
python3 setup.py install
147
136
148
137
WORKDIR /tmp
149
- COPY ./requirements.txt /tmp
138
+ COPY ./requirements.21.10. txt /tmp/requirements.txt
150
139
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir --ignore-installed -r requirements.txt && \
151
140
rm -f /tmp/*.whl /tmp/requirements.txt
152
141
@@ -235,7 +224,7 @@ LABEL ai.backend.kernelspec="1" \
235
224
ai.backend.resource.min.cpu="1" \
236
225
ai.backend.resource.min.mem="1g" \
237
226
ai.backend.resource.min.cuda.device=1 \
238
- ai.backend.resource.min.cuda.shares=0.1 \
227
+ ai.backend.resource.min.cuda.shares=0 \
239
228
ai.backend.base-distro="ubuntu16.04" \
240
229
ai.backend.runtime-type="python" \
241
230
ai.backend.runtime-path="/opt/conda/bin/python3" \
0 commit comments