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:5157f3aaa0491a3797937b21051e97da9e545f3a2f335aa43f06eeb0dae8d4c4RUN
OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz OPENCV_URL=https://nvidia.box.com/shared/static/2hssa5g3v28ozvo3tc3qwxmn78yerca9.gz PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install --no-cache-dir --verbose pycuda six
08/04/2022 6:08 PM UTC
sha256:5d1a39e7fb75170231f4394299c2dde94ed835567756ddd76d235eab80075f5dRUN
OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz OPENCV_URL=https://nvidia.box.com/shared/static/2hssa5g3v28ozvo3tc3qwxmn78yerca9.gz PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
08/04/2022 6:01 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64:
08/04/2022 6:01 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
PATH=/usr/local/cuda/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
08/04/2022 6:01 PM UTC
sha256:0e651afae557b766ce9801749b4198778a495ca74463a884c2d343b4c50ce659RUN
OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz OPENCV_URL=https://nvidia.box.com/shared/static/2hssa5g3v28ozvo3tc3qwxmn78yerca9.gz PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 cd /tmp &&
  ./opencv_install.sh ${OPENCV_URL} ${OPENCV_DEB}
08/04/2022 6:01 PM UTC
sha256:fe175acea3d15f2b2f536d48a504dd9e0eed366dd672214274df72e6ac330e4cCOPY
file:738b23d3effc8e9ffc54a87257f6eadd655f97205a53b370e43cf91ac36a2748 in /tmp/opencv_install.sh
08/04/2022 6:00 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz
08/04/2022 6:00 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
OPENCV_URL=https://nvidia.box.com/shared/static/5v89u6g5rb62fpz4lh0rz531ajo2t5ef.gz
08/04/2022 6:00 PM UTC
sha256:7fea22c659c1632b4b41541b3bbf1910dde703a7abc3ee42abcb671e99c94faaRUN
PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 git clone --recursive -b ${TORCHAUDIO_VERSION} https://github.com/pytorch/audio torchaudio &&
  cd torchaudio &&
  sed -i 's#  URL https://zlib.net/zlib-1.2.11.tar.gz#  URL https://zlib.net/zlib-1.2.12.tar.gz#g' third_party/zlib/CMakeLists.txt ||
  echo "failed to patch torchaudio/third_party/zlib/CMakeLists.txt" &&
  sed -i 's#  URL_HASH SHA256=c3e5e9fdd5004dcb542feda5ee4f0ff0744628baf8ed2dd5d66f8ca1197cb1a1#  URL_HASH SHA256=91844808532e5ce316b3c010929493c0244f3d37593afd6de04f71821d5136d9#g' third_party/zlib/CMakeLists.txt ||
  echo "failed to patch torchaudio/third_party/zlib/CMakeLists.txt" &&
  BUILD_SOX=1 python3 setup.py install &&
  cd ../ &&
  rm -rf torchaudio
08/04/2022 6:00 PM UTC
sha256:834bf6dfd9568f0ccdb7ef65e3fac86e1b771a2eb2ca96635b72b2a691748b61RUN
PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install --no-cache-dir scikit-build &&
  pip3 install --no-cache-dir ninja &&
  pip3 install --no-cache-dir --verbose pysoundfile
08/04/2022 5:40 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.