Meta
NVIDIA L4T PyTorch
Container
Meta
NVIDIA L4T PyTorch

PyTorch is a GPU accelerated tensor computational framework with a Python front end. This container contains PyTorch and torchvision pre-installed in a Python 3 environment to get up & running quickly with PyTorch on Jetson.

  • LayerLabelCreated
    sha256:88f584b348f7e1373e412f66d0920bc7492a187320b931eeb0303c7b63de51b3RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/3ibazbiwtkl181n95n9em3wtrca7tdzp.whl PYTORCH_WHL=torch-1.5.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.6.0 pip3 install pycuda --verbose
    04/27/2020 7:31 PM UTC
    sha256:3946529dd4e734918330ea55cf97bc9fc2619f0363b1b027633b53ea507dbdbdRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/3ibazbiwtkl181n95n9em3wtrca7tdzp.whl PYTORCH_WHL=torch-1.5.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.6.0 echo "$PATH" &&
      echo "$LD_LIBRARY_PATH"
    04/27/2020 7:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    04/27/2020 7:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PATH=/usr/local/cuda/bin:/usr/local/cuda-10.2/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
    04/27/2020 7:25 PM UTC
    sha256:6b937cb4986ced6f9f0467194854616b30f99d63a76630705fb112ee8520cdbaRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/3ibazbiwtkl181n95n9em3wtrca7tdzp.whl PYTORCH_WHL=torch-1.5.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.6.0 git clone -b ${TORCHVISION_VERSION} https://github.com/pytorch/vision torchvision &&
      cd torchvision &&
      python3 setup.py install &&
      cd ../ &&
      rm -rf torchvision &&
      pip3 install "${PILLOW_VERSION}"
    04/27/2020 7:25 PM UTC
    sha256:a1f50a89a59c8210578a99bf6f204b1a242aea92d2a55e14c6b79d970a28fbb8RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/3ibazbiwtkl181n95n9em3wtrca7tdzp.whl PYTORCH_WHL=torch-1.5.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.6.0 apt-get update &&
      apt-get install -y --no-install-recommends git build-essential libjpeg-dev zlib1g-dev &&
      rm -rf /var/lib/apt/lists/*
    04/27/2020 7:11 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    PILLOW_VERSION=pillow<7
    04/27/2020 7:11 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHVISION_VERSION=v0.4.0
    04/27/2020 7:11 PM UTC
    sha256:d20488b99ffa4281fb601b9309589620d7deb0aee17097b22805c6fed9d9ad6aRUN
    PYTORCH_URL=https://nvidia.box.com/shared/static/3ibazbiwtkl181n95n9em3wtrca7tdzp.whl PYTORCH_WHL=torch-1.5.0-cp36-cp36m-linux_aarch64.whl wget --quiet --show-progress --progress=bar:force:noscroll --no-check-certificate ${PYTORCH_URL} -O ${PYTORCH_WHL} &&
      pip3 install ${PYTORCH_WHL} --verbose &&
      rm ${PYTORCH_WHL}
    04/27/2020 7:11 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    PYTORCH_WHL=torch-1.2.0-cp36-cp36m-linux_aarch64.whl
    04/27/2020 6:49 PM UTC