File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -9,14 +9,15 @@ ENV NCCL=2.8.3
9
9
ENV NCCL_VERSION=2.8.3-1+cuda11.1
10
10
ENV CUDNN_VERSION 8.0.5.43
11
11
ENV MLNX_OFED_VERSION=5.1-2.3.7.1
12
+ # Intel MKL
12
13
ENV MKL_VERSION=2020.3
13
14
ENV MKL_BUILD=279
14
- ENV NUMPY_VERSION 1.19.4
15
- ENV SCIPY_VERSION 1.5.4
16
- ENV OPENMPI 4.0.5
17
15
# Python 3.8 is supported by Ubuntu Bionic out of the box
18
16
ARG python=3.8
19
17
ENV PYTHON_VERSION=${python}
18
+ ENV NUMPY_VERSION 1.19.4
19
+ ENV SCIPY_VERSION 1.5.4
20
+ ENV OPENMPI 4.0.5
20
21
21
22
ENV DEBIAN_FRONTEND=noninteractive \
22
23
MPLBACKEND=Svg \
@@ -190,9 +191,6 @@ RUN python3 -m pip install --no-cache-dir \
190
191
nvidia-pyindex==1.0.5 \
191
192
matplotlib==3.3.3
192
193
193
- # pystan==2.19.1.1 \
194
- # pycairo==1.12.0 \
195
- # python-language-server[all] \
196
194
197
195
WORKDIR /tmp
198
196
RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git && \
You can’t perform that action at this time.
0 commit comments