—

/bin/bash -c #(nop) ENTRYPOINT []

/bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}

HUGECTR_VER=v3.2 RELEASE=true /bin/bash -c echo $(du -h --max-depth=1 /)

HUGECTR_VER=v3.2 RELEASE=true /bin/bash -c pip install numba numpy --upgrade

HUGECTR_VER=v3.2 RELEASE=true /bin/bash -c rm -rf /repos

/bin/bash -c #(nop) ENV PYTHONPATH=:/usr/local/hugectr/lib

/bin/bash -c #(nop) ENV LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/lib:/repos/dist/lib:/usr/local/hugectr/lib PATH=/usr/local/cuda/lib64/:/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/cuda/bin:/usr/lib/x86_64-linux-gnu/:/usr/local/hugectr/bin

HUGECTR_VER=v3.2 RELEASE=true /bin/bash -c rm /usr/local/cuda/lib64/stubs/libcuda.so.1

HUGECTR_VER=v3.2 RELEASE=true /bin/bash -c cd /var/tmp/HugeCTR &&
  mkdir build &&
  cd build &&
  LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH &&
  export PATH=$PATH:/usr/local/cuda-${CUDA_SHORT_VERSION}/compat/ &&
  cmake -DCMAKE_CXX_COMPILER=/usr/bin/g++ -DCMAKE_C_COMPILER=/usr/bin/gcc -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DVAL_MODE=OFF -DENABLE_MULTINODES=ON -DNCCL_A2A=ON .. &&
  make VERBOSE=1 -j$(nproc) &&
  make install &&
  chmod +x /usr/local/hugectr/bin/* &&
  chmod +x /usr/local/hugectr/lib/* &&
  cd /var/tmp/HugeCTR/onnx_converter &&
  python3 setup.py install &&
  rm -rf /var/tmp/HugeCTR