Skip to main content
NVIDIA
TAO Toolkit
Container
NVIDIA
TAO Toolkit

Docker containers distributed as part of the TAO Toolkit package

  • LayerLabelCreated
    sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
    /workspace/
    08/21/2024 7:04 PM UTC
    sha256:70af34c8227be1bd4fc800e9b30111e5eedc43edf7964f494a89fa9bb1ae837dRUN
    user_id=1000 user_name=developer groups=developer:1000 home=/home/developer echo "ALL   ALL = (ALL) NOPASSWD: ALL" > /etc/sudoers &&
      mkdir -p "$(dirname $home)"
    08/21/2024 7:04 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    home=/home/developer
    08/21/2024 7:04 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    groups=developer:1000
    08/21/2024 7:04 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    user_name=developer
    08/21/2024 7:04 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    user_id=1000
    08/21/2024 7:04 PM UTC
    sha256:d3a40a59e6a111c301b2b963a6b031e4b966d8d1d551b0fc0fafb5d000a8cbd2RUN
    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
    08/21/2024 7:04 PM UTC
    sha256:b457582535d02913104857a3d2dcb3ee5fec9de3083675021c1317abb75d838bCOPY
    third_party/mmcv/mmcv.patch mmcv.patch
    08/21/2024 6:57 PM UTC
    sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
    /opt
    08/21/2024 6:57 PM UTC
    sha256:d40259e67952b9dcc2358ae576bcbfda900db3af755936c15328cf3008869969RUN
    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 *
    08/21/2024 6:57 PM UTC
    ...