Skip to content

Commit c1f6d00

Browse files
committed
clean up packages
1 parent 87b3e1d commit c1f6d00

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

commons/Dockerfile.base.py38-cuda11.1

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,15 @@ ENV NCCL=2.8.3
99
ENV NCCL_VERSION=2.8.3-1+cuda11.1
1010
ENV CUDNN_VERSION 8.0.5.43
1111
ENV MLNX_OFED_VERSION=5.1-2.3.7.1
12+
# Intel MKL
1213
ENV MKL_VERSION=2020.3
1314
ENV MKL_BUILD=279
14-
ENV NUMPY_VERSION 1.19.4
15-
ENV SCIPY_VERSION 1.5.4
16-
ENV OPENMPI 4.0.5
1715
# Python 3.8 is supported by Ubuntu Bionic out of the box
1816
ARG python=3.8
1917
ENV PYTHON_VERSION=${python}
18+
ENV NUMPY_VERSION 1.19.4
19+
ENV SCIPY_VERSION 1.5.4
20+
ENV OPENMPI 4.0.5
2021

2122
ENV DEBIAN_FRONTEND=noninteractive \
2223
MPLBACKEND=Svg \
@@ -190,9 +191,6 @@ RUN python3 -m pip install --no-cache-dir \
190191
nvidia-pyindex==1.0.5 \
191192
matplotlib==3.3.3
192193

193-
# pystan==2.19.1.1 \
194-
# pycairo==1.12.0 \
195-
# python-language-server[all] \
196194

197195
WORKDIR /tmp
198196
RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git && \

0 commit comments

Comments
 (0)