Skip to content

Commit 091cb34

Browse files
committed
add base kernel 20.08 for python 3.6 & CUDA10.2
1 parent c343a73 commit 091cb34

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

commons/Dockerfile.base.20.08-py36-cuda10.1

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -173,9 +173,20 @@ RUN python3 -m pip install --no-cache-dir --upgrade -Iv \
173173
matplotlib==3.1.1
174174

175175
WORKDIR /tmp
176-
RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git && \
177-
cd OpenBLAS && \
178-
make DYNAMIC_ARCH=1 NO_AFFINITY=1 NUM_THREADS=48 FC=gfortran && \
176+
RUN mkdir -p /tmp/objconv && \
177+
cd /tmp/objconv && \
178+
curl -O "https://raw.githubusercontent.com/MacPython/openblas-libs/master/objconv/source.zip" && \
179+
unzip source.zip && \
180+
g++ -O1 -o objconv *.cpp
181+
WORKDIR /tmp/openblas
182+
ENV CFLAGS=-Wl,-strip-all \
183+
CXXFLAGS=-Wl,-strip-all \
184+
FFLAGS=-Wl,-strip-all
185+
186+
RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git /tmp/openblas && \
187+
cd /tmp/openblas && \
188+
make DYNAMIC_ARCH=1 NO_AFFINITY=1 USE_OPENMP=0 NUM_THREADS=64 BINARY=64 INTERFACE64=1 SYMBOLSUFFIX=64_ OBJCONV=/tmp/objconv/objconv FC=gfortran > /dev/null 2>&1 && \
189+
make PREFIX=/opt/OpenBLAS INTERFACE64=1 SYMBOLSUFFIX=64_ OBJCONV=/tmp/objconv/objconv install && \
179190
make install
180191
RUN git clone --recursive https://github.com/bodono/scs-python.git && \
181192
cd /tmp/scs-python && \

0 commit comments

Comments
 (0)