@@ -5,11 +5,12 @@ ENV DEBIAN_FRONTEND=noninteractive \
5
5
MPLBACKEND=Svg \
6
6
PIP_IGNORE_INSTALLED=0 \
7
7
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 " \
10
10
mecab_dicdir=/usr/local/lib/mecab/dic/mecab-ko-dic \
11
11
CPLUS_INCLUDE_PATH=/usr/include/gdal \
12
12
C_INCLUDE_PATH=/usr/include/gdal \
13
+ CPATH=/usr/local/cuda-10.1/targets/x86_64-linux/include:$CPATH \
13
14
LANG=C.UTF-8
14
15
15
16
RUN apt-get update && \
@@ -57,10 +58,18 @@ RUN apt-get update && \
57
58
x264
58
59
59
60
# 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 && \
61
62
rm -rf /var/lib/apt/lists/* && \
62
63
ldconfig
63
64
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
+
64
73
# nvtop install
65
74
WORKDIR /tmp
66
75
RUN git clone https://github.com/Syllo/nvtop.git && \
@@ -98,7 +107,7 @@ RUN echo "Install mecab-ko-dic" && \
98
107
git clone https://bitbucket.org/eunjeon/mecab-python-0.996.git && \
99
108
python3 -m pip install /tmp/mecab-python-0.996
100
109
101
- RUN curl -sL https://deb.nodesource.com/setup_14 .x | bash - && \
110
+ RUN curl -sL https://deb.nodesource.com/setup_16 .x | bash - && \
102
111
apt-get update -y && \
103
112
apt-get install -y nodejs
104
113
@@ -157,10 +166,11 @@ RUN mkdir /tmp/openmpi && \
157
166
echo "btl_tcp_if_exclude = lo,docker0" >> /usr/local/etc/openmpi-mca-params.conf
158
167
159
168
# 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 && \
161
171
HOROVOD_CUDA_HOME=$CONDA_PREFIX HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
162
172
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 && \
164
174
ldconfig
165
175
166
176
RUN python3 -m pip install --no-cache-dir \
0 commit comments