Loading pytorchimage-centos-cuda/Dockerfile +13 −12 Original line number Diff line number Diff line Loading @@ -167,19 +167,20 @@ RUN cd ${INSTALL_ROOT} \ && cd ${INSTALL_ROOT} \ && rm -rf opencv # Install horovod ## Install DALI (not working) COPY ./cmake_patch /code RUN cd ${INSTALL_ROOT} \ && git clone --recurse-submodules --shallow-submodules https://github.com/NVIDIA/DALI --depth 1 --branch v1.6.0 \ && cd DALI \ && git apply /code/cmake_patch \ && mkdir build \ && cd build \ && ln -s /usr/lib64/libclang.so /usr/lib64/libclang-11.so \ && cmake -DCMAKE_BUILD_TYPE=Release -DBUILD_NVJPEG2K=OFF -DFFMPEG_ROOT_DIR=${CONDA_HOME} -DJPEG_INCLUDE_DIR=/usr/local/include -DJPEG_LIBRARY=/usr/local/lib/libturbojpeg.so -DCUDA_TARGET_ARCHS=70 -DBUILD_PROTOBUF=ON -DProtobuf_LIBRARY=/usr/local/lib/libprotobuf.a -DFFmpeg_Libavcodec=/usr/local/lib/libavcodec.so -DFFmpeg_Libavfilter=/usr/local/lib/libavfilter.so -DFFmpeg_Libavformat=/usr/local/lib/libavformat.so -DFFmpeg_Libavutil=/usr/local/lib/libavutil.so .. \ && make -j1 \ && make install \ && pip install dali/python #COPY ./cmake_patch /code #RUN cd ${INSTALL_ROOT} \ # && git clone --recurse-submodules --shallow-submodules https://github.com/NVIDIA/DALI --depth 1 --branch v1.6.0 \ # && cd DALI \ # && git apply /code/cmake_patch \ # && mkdir build \ # && cd build \ # && ln -s /usr/lib64/libclang.so /usr/lib64/libclang-11.so \ # && cmake -DCMAKE_BUILD_TYPE=Release -DBUILD_NVJPEG2K=OFF -DFFMPEG_ROOT_DIR=${CONDA_HOME} -DJPEG_INCLUDE_DIR=/usr/local/include -DJPEG_LIBRARY=/usr/local/lib/libturbojpeg.so -DCUDA_TARGET_ARCHS=70 -DBUILD_PROTOBUF=ON -DProtobuf_LIBRARY=/usr/local/lib/libprotobuf.a -DFFmpeg_Libavcodec=/usr/local/lib/libavcodec.so -DFFmpeg_Libavfilter=/usr/local/lib/libavfilter.so -DFFmpeg_Libavformat=/usr/local/lib/libavformat.so -DFFmpeg_Libavutil=/usr/local/lib/libavutil.so .. \ # && make -j1 \ # && make install \ # && pip install dali/python ### failed attempt 1 #RUN cd ${INSTALL_ROOT} \ Loading Loading
pytorchimage-centos-cuda/Dockerfile +13 −12 Original line number Diff line number Diff line Loading @@ -167,19 +167,20 @@ RUN cd ${INSTALL_ROOT} \ && cd ${INSTALL_ROOT} \ && rm -rf opencv # Install horovod ## Install DALI (not working) COPY ./cmake_patch /code RUN cd ${INSTALL_ROOT} \ && git clone --recurse-submodules --shallow-submodules https://github.com/NVIDIA/DALI --depth 1 --branch v1.6.0 \ && cd DALI \ && git apply /code/cmake_patch \ && mkdir build \ && cd build \ && ln -s /usr/lib64/libclang.so /usr/lib64/libclang-11.so \ && cmake -DCMAKE_BUILD_TYPE=Release -DBUILD_NVJPEG2K=OFF -DFFMPEG_ROOT_DIR=${CONDA_HOME} -DJPEG_INCLUDE_DIR=/usr/local/include -DJPEG_LIBRARY=/usr/local/lib/libturbojpeg.so -DCUDA_TARGET_ARCHS=70 -DBUILD_PROTOBUF=ON -DProtobuf_LIBRARY=/usr/local/lib/libprotobuf.a -DFFmpeg_Libavcodec=/usr/local/lib/libavcodec.so -DFFmpeg_Libavfilter=/usr/local/lib/libavfilter.so -DFFmpeg_Libavformat=/usr/local/lib/libavformat.so -DFFmpeg_Libavutil=/usr/local/lib/libavutil.so .. \ && make -j1 \ && make install \ && pip install dali/python #COPY ./cmake_patch /code #RUN cd ${INSTALL_ROOT} \ # && git clone --recurse-submodules --shallow-submodules https://github.com/NVIDIA/DALI --depth 1 --branch v1.6.0 \ # && cd DALI \ # && git apply /code/cmake_patch \ # && mkdir build \ # && cd build \ # && ln -s /usr/lib64/libclang.so /usr/lib64/libclang-11.so \ # && cmake -DCMAKE_BUILD_TYPE=Release -DBUILD_NVJPEG2K=OFF -DFFMPEG_ROOT_DIR=${CONDA_HOME} -DJPEG_INCLUDE_DIR=/usr/local/include -DJPEG_LIBRARY=/usr/local/lib/libturbojpeg.so -DCUDA_TARGET_ARCHS=70 -DBUILD_PROTOBUF=ON -DProtobuf_LIBRARY=/usr/local/lib/libprotobuf.a -DFFmpeg_Libavcodec=/usr/local/lib/libavcodec.so -DFFmpeg_Libavfilter=/usr/local/lib/libavfilter.so -DFFmpeg_Libavformat=/usr/local/lib/libavformat.so -DFFmpeg_Libavutil=/usr/local/lib/libavutil.so .. \ # && make -j1 \ # && make install \ # && pip install dali/python ### failed attempt 1 #RUN cd ${INSTALL_ROOT} \ Loading