Skip to content

Commit 0b384a9

Browse files
committed
add OPenBlas
1 parent aa0aedb commit 0b384a9

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

vendor/ngc-pytorch/Dockerfile.20.12-py3

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
FROM nvcr.io/nvidia/pytorch:20.12-py3
2-
# NVIDIA PyTorch with Python 3.6 (CONDA)
2+
# NVIDIA PyTorch with Python 3.8 (CONDA)
33

44

55

@@ -80,7 +80,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
8080
rm -rf /var/lib/apt/lists/
8181

8282
WORKDIR /tmp
83-
ENV OPENCV_VERSION="4.5.0"
83+
ENV OPENCV_VERSION="4.5.1"
8484
RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
8585
wget -O opencv-contrib.zip https://github.com/opencv/opencv_contrib/archive/${OPENCV_VERSION}.zip && \
8686
unzip ${OPENCV_VERSION}.zip && \
@@ -105,7 +105,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
105105
-D WITH_V4L=ON \
106106
-D BUILD_TESTS=OFF \
107107
-D BUILD_PERF_TESTS=OFF \
108-
-D OPENCV_EXTRA_MODULES_PATH="../../opencv_contrib-4.5.0/modules" \
108+
-D OPENCV_EXTRA_MODULES_PATH="../../opencv_contrib-4.5.1/modules" \
109109
-D CMAKE_BUILD_TYPE=RELEASE \
110110
-D CMAKE_INSTALL_PREFIX=$(python3 -c "import sys; print(sys.prefix)") \
111111
-D PYTHON_EXECUTABLE=$(which python3) \
@@ -126,6 +126,10 @@ RUN curl -sL https://deb.nodesource.com/setup_10.x | bash - && \
126126
apt-get install -y nodejs
127127

128128
WORKDIR /tmp
129+
RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git && \
130+
cd OpenBLAS && \
131+
make DYNAMIC_ARCH=1 NO_AFFINITY=1 NUM_THREADS=48 FC=gfortran && \
132+
make install
129133
RUN git clone --recursive https://github.com/bodono/scs-python.git && \
130134
cd /tmp/scs-python && \
131135
python setup.py install --scs --gpu
@@ -247,7 +251,7 @@ LABEL ai.backend.kernelspec="1" \
247251
ai.backend.base-distro="ubuntu16.04" \
248252
ai.backend.runtime-type="python" \
249253
ai.backend.runtime-path="/opt/conda/bin/python3" \
250-
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8091,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:http:5000,nniboard:http:8080"
254+
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8091,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:preopen:5000,nniboard:preopen:8080"
251255

252256
WORKDIR /home/work
253257
# vim: ft=dockerfile

0 commit comments

Comments
 (0)