Meta
Meta
NVIDIA L4T PyTorch
Container
Meta
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:b9e9d3887268d964dc030981816b451786d6c21b5ccd923d3d733afab493f616RUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install pycuda --verbose
12/07/2020 9:39 PM UTC
sha256:c8219a7d4bea8d989e4733ec9ff71a8fbe2b5e5398cc3a738113b0036128de93RUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
12/07/2020 9:35 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
12/07/2020 9:35 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
12/07/2020 9:35 PM UTC
sha256:93b7b1478d26a17d7ba5678af4d6fd39f309db25f4b1ae8a7156a589e93ff286RUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 git clone -b ${TORCHAUDIO_VERSION} https://github.com/pytorch/audio torchaudio &&
  cd torchaudio &&
  python3 setup.py install &&
  cd ../ &&
  rm -rf torchaudio
12/07/2020 9:35 PM UTC
sha256:1885dddc79e815e731b35aa42197ec8c2776e01b72ed68a135046a805e77956fRUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 apt-get update &&
  apt-get install -y --no-install-recommends sox libsox-dev libsox-fmt-all &&
  rm -rf /var/lib/apt/lists/*
12/07/2020 9:31 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
TORCHAUDIO_VERSION
12/07/2020 9:31 PM UTC
sha256:561e500c4b69ca6a99924382e8fc3b42d2512864bc0deebf9adfe5d5831f7462RUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 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}"
12/07/2020 9:31 PM UTC
sha256:e429652ff20241e7a58695362b57fa365d0debae9a61c63026d86b4c0c7795d3RUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 apt-get update &&
  apt-get install -y --no-install-recommends git build-essential libjpeg-dev zlib1g-dev &&
  rm -rf /var/lib/apt/lists/*
12/07/2020 9:19 PM UTC
sha256:786c90dad7299cc52d0dea478d51565337e641dac1efd54141fa7c65eac4b8c2RUN
PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 printenv &&
  echo "torchvision version = $TORCHVISION_VERSION" &&
  echo "pillow version = $PILLOW_VERSION" &&
  echo "TORCH_CUDA_ARCH_LIST = $TORCH_CUDA_ARCH_LIST"
12/07/2020 9:19 PM UTC

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.