1
1
FROM nvcr.io/nvidia/pytorch:20.12-py3
2
- # NVIDIA PyTorch with Python 3.6 (CONDA)
2
+ # NVIDIA PyTorch with Python 3.8 (CONDA)
3
3
4
4
5
5
@@ -80,7 +80,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
80
80
rm -rf /var/lib/apt/lists/
81
81
82
82
WORKDIR /tmp
83
- ENV OPENCV_VERSION="4.5.0 "
83
+ ENV OPENCV_VERSION="4.5.1 "
84
84
RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
85
85
wget -O opencv-contrib.zip https://github.com/opencv/opencv_contrib/archive/${OPENCV_VERSION}.zip && \
86
86
unzip ${OPENCV_VERSION}.zip && \
@@ -105,7 +105,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
105
105
-D WITH_V4L=ON \
106
106
-D BUILD_TESTS=OFF \
107
107
-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" \
109
109
-D CMAKE_BUILD_TYPE=RELEASE \
110
110
-D CMAKE_INSTALL_PREFIX=$(python3 -c "import sys; print(sys.prefix)" ) \
111
111
-D PYTHON_EXECUTABLE=$(which python3) \
@@ -126,6 +126,10 @@ RUN curl -sL https://deb.nodesource.com/setup_10.x | bash - && \
126
126
apt-get install -y nodejs
127
127
128
128
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
129
133
RUN git clone --recursive https://github.com/bodono/scs-python.git && \
130
134
cd /tmp/scs-python && \
131
135
python setup.py install --scs --gpu
@@ -247,7 +251,7 @@ LABEL ai.backend.kernelspec="1" \
247
251
ai.backend.base-distro="ubuntu16.04" \
248
252
ai.backend.runtime-type="python" \
249
253
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"
251
255
252
256
WORKDIR /home/work
253
257
# vim: ft=dockerfile
0 commit comments