Skip to content

Commit a19ebe0

Browse files
committed
update nodejs
1 parent 71b4731 commit a19ebe0

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

vendor/ngc-pytorch/Dockerfile.22.02-py3

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,12 @@ ENV DEBIAN_FRONTEND=noninteractive \
55
MPLBACKEND=Svg \
66
PIP_IGNORE_INSTALLED=0 \
77
PYTHONUNBUFFERED=1 \
8-
LD_LIBRARY_PATH="/usr/local/cuda/compat/lib:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
9-
PATH="/usr/local/nvm/versions/node/v16.14.0/bin:/opt/conda/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin:/usr/local/src/lightgbm/LightGBM:/usr/local/bin/mecab" \
8+
LD_LIBRARY_PATH="/usr/local/cuda/compat/lib:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/local/cuda-11.6/include:/usr/include/x86_64-linux-gnu:$LD_LIBRARY_PATH" \
9+
PATH="/usr/local/nvm/versions/node/v16.6.1/bin:/opt/conda/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/cuda-11.6/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin:/usr/local/src/lightgbm/LightGBM:/usr/local/bin/mecab:$PATH" \
1010
mecab_dicdir=/usr/local/lib/mecab/dic/mecab-ko-dic \
1111
CPLUS_INCLUDE_PATH=/usr/include/gdal \
1212
C_INCLUDE_PATH=/usr/include/gdal \
13+
CPATH=/usr/local/cuda-10.1/targets/x86_64-linux/include:$CPATH \
1314
LANG=C.UTF-8
1415

1516
RUN apt-get update && \
@@ -57,10 +58,18 @@ RUN apt-get update && \
5758
x264
5859

5960
# Install CUDA + cuDNN
60-
RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.2.4 /usr/local/cuda/lib64/libcudnn.so && \
61+
RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.3.2 /usr/local/cuda/lib64/libcudnn.so && \
6162
rm -rf /var/lib/apt/lists/* && \
6263
ldconfig
6364

65+
# Install cmake
66+
RUN wget https://github.com/Kitware/CMake/releases/download/v3.22.0/cmake-3.22.0-Linux-x86_64.sh \
67+
-q -O /tmp/cmake-install.sh && \
68+
chmod u+x /tmp/cmake-install.sh && \
69+
mkdir /usr/bin/cmake && \
70+
/tmp/cmake-install.sh --skip-license --prefix=/usr/bin/cmake && \
71+
rm /tmp/cmake-install.sh
72+
6473
# nvtop install
6574
WORKDIR /tmp
6675
RUN git clone https://github.com/Syllo/nvtop.git && \
@@ -98,7 +107,7 @@ RUN echo "Install mecab-ko-dic" && \
98107
git clone https://bitbucket.org/eunjeon/mecab-python-0.996.git && \
99108
python3 -m pip install /tmp/mecab-python-0.996
100109

101-
RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
110+
RUN curl -sL https://deb.nodesource.com/setup_16.x | bash - && \
102111
apt-get update -y && \
103112
apt-get install -y nodejs
104113

@@ -157,10 +166,11 @@ RUN mkdir /tmp/openmpi && \
157166
echo "btl_tcp_if_exclude = lo,docker0" >> /usr/local/etc/openmpi-mca-params.conf
158167

159168
# Install Horovod, temporarily using CUDA stubs
160-
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
169+
RUN cp /usr/local/cuda-11.6/bin/nvcc /bin/nvcc && \
170+
# ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
161171
HOROVOD_CUDA_HOME=$CONDA_PREFIX HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
162172
HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1 \
163-
pip install --no-cache-dir horovod==0.23.0 && \
173+
pip install --no-cache-dir horovod==0.24.1 && \
164174
ldconfig
165175

166176
RUN python3 -m pip install --no-cache-dir \

0 commit comments

Comments
 (0)