Skip to content

Commit 8b3b67b

Browse files
committed
update cuda version
1 parent 350f29f commit 8b3b67b

File tree

1 file changed

+17
-14
lines changed

1 file changed

+17
-14
lines changed

commons/Dockerfile.base.py38-cuda11.2

Lines changed: 17 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -238,8 +238,8 @@ RUN mkdir -p /opt/oracle && \
238238

239239
# install git-lfs
240240
WORKDIR /tmp
241-
RUN curl -sLO https://github.com/git-lfs/git-lfs/releases/download/v3.2.0/git-lfs-linux-amd64-v3.2.0.tar.gz && \
242-
tar -zxf git-lfs-linux-amd64-v3.2.0.tar.gz && \
241+
RUN curl -sLO https://github.com/git-lfs/git-lfs/releases/download/v3.0.2/git-lfs-linux-amd64-v3.0.2.tar.gz && \
242+
tar -zxf git-lfs-linux-amd64-v3.0.2.tar.gz && \
243243
bash install.sh && \
244244
rm -rf /tmp/*
245245

@@ -262,7 +262,7 @@ RUN wget https://github.com/openucx/ucx/archive/v1.12.1.tar.gz && \
262262
make install
263263

264264
# OpenBlas
265-
RUN git clone -q --branch=v0.3.20 git://github.com/xianyi/OpenBLAS.git && \
265+
RUN git clone -q --branch=v0.3.20 https://github.com/xianyi/OpenBLAS.git && \
266266
cd OpenBLAS && \
267267
make DYNAMIC_ARCH=1 NO_AFFINITY=1 NUM_THREADS=48 FC=gfortran && \
268268
make install
@@ -271,19 +271,12 @@ RUN git clone -q --branch=v0.3.20 git://github.com/xianyi/OpenBLAS.git && \
271271
WORKDIR /tmp
272272
RUN curl https://bootstrap.pypa.io/get-pip.py | python3 && \
273273
# rm -r /usr/bin/python3 && \
274-
python3 -m pip install --no-cache-dir -U setuptools pip && \
274+
python3 -m pip install --no-cache-dir -U setuptools "pip < 22" && \
275275
python3 -m pip install --no-cache-dir cython pybind11 pythran && \
276276
# ln -s /usr/bin/python3.8 /usr/bin/python3 && \
277277
update-alternatives --install /usr/bin/python python /usr/bin/python3 1 && \
278278
update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 2
279279

280-
RUN python3 -m pip install --extra-index-url \
281-
https://developer.download.nvidia.com/compute/redist \
282-
--upgrade nvidia-dali-cuda110 \
283-
&& \
284-
python3 -m pip install --extra-index-url \
285-
https://developer.download.nvidia.com/compute/redist \
286-
--upgrade nvidia-dali-tf-plugin-cuda110
287280

288281
# XGBoost
289282
RUN git clone --recursive https://github.com/dmlc/xgboost && \
@@ -304,6 +297,7 @@ RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
304297
cd /usr/local/src/lightgbm/LightGBM/python-package && \
305298
python3 setup.py install --precompile
306299

300+
ENV NUMPY_VERSION=1.17.4
307301
# Numpy
308302
RUN git clone --branch=v${NUMPY_VERSION} --depth=1 https://github.com/numpy/numpy.git numpy && \
309303
cd numpy && \
@@ -317,6 +311,7 @@ RUN git clone --branch=v${NUMPY_VERSION} --depth=1 https://github.com/numpy/nump
317311
echo "lapack_libs =" >> site.cfg && \
318312
python3 setup.py bdist_wheel -d /tmp
319313

314+
ENV SCIPY_VERSION=1.6.0
320315
# Scipy
321316
RUN cd /tmp && \
322317
git clone --branch=v${SCIPY_VERSION} --depth=1 https://github.com/scipy/scipy.git scipy && \
@@ -361,9 +356,9 @@ RUN curl -LO https://bitbucket.org/eunjeon/mecab-ko/downloads/mecab-0.996-ko-0.9
361356
python3 -m pip install /tmp/mecab-python-0.996
362357

363358
WORKDIR /tmp
364-
COPY ./requirements.py38.txt /tmp
365-
RUN python3 -m pip install --no-cache-dir -r requirements.py38.txt && \
366-
rm -f /tmp/*.whl /tmp/requirements.py38.txt
359+
COPY ./requirements.22.06.txt /tmp/requirements.txt
360+
RUN python3 -m pip install --no-cache-dir -r requirements.txt && \
361+
rm -f /tmp/*.whl /tmp/requirements.txt
367362

368363
# OpenCV
369364
WORKDIR /tmp
@@ -403,6 +398,14 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
403398
make -j$(nproc) && \
404399
make install
405400

401+
RUN python3 -m pip install --extra-index-url \
402+
https://developer.download.nvidia.com/compute/redist \
403+
--upgrade nvidia-dali-cuda110 \
404+
&& \
405+
python3 -m pip install --extra-index-url \
406+
https://developer.download.nvidia.com/compute/redist \
407+
--upgrade nvidia-dali-tf-plugin-cuda110
408+
406409
# Install Open MPI
407410
RUN mkdir /tmp/openmpi && \
408411
cd /tmp/openmpi && \

0 commit comments

Comments
 (0)