Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/; ExposedPorts 6006/tcp, 8888/tcp

user_id=1000 user_name=developer groups=developer:1000 home=/home/developer echo "ALL   ALL = (ALL) NOPASSWD: ALL" > /etc/sudoers &&
  mkdir -p "$(dirname $home)"

home=/home/developer

groups=developer:1000

user_name=developer

user_id=1000

git clone https://github.com/open-mmlab/mmcv.git &&
  cd mmcv &&
  git checkout v1.7.1 &&
  git apply /opt/mmcv.patch &&
  pip install -r requirements/optional.txt --ignore-installed &&
  FORCE_CUDA=1 MMCV_WITH_OPS=1 python setup.py install

third_party/mmcv/mmcv.patch mmcv.patch

/opt

export WORKSPACE=/workspace/tlt-pytorch &&
  cd ${WORKSPACE} &&
  bash release/docker/build_wheel.sh &&
  find dist/ -name "nvidia_tao_pytorch*.whl" -type f | xargs -n 1 pip install &&
  rm -rf *