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:452429c910e118e79f06b4af917a530824da693d351b28334f49e254cffada8dRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl PYTORCH_WHL=torch-1.4.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.5.0 pip3 install pycuda --verbose
    04/27/2020 7:09 PM UTC
    sha256:6932ab50b9faee84a54ac3b36bb1cbd0a111ac66476ae5c48342a99cd4731a8dRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl PYTORCH_WHL=torch-1.4.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.5.0 echo "$PATH" &&
      echo "$LD_LIBRARY_PATH"
    04/27/2020 7:03 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    04/27/2020 7:03 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:03 PM UTC
    sha256:8760ad2a761c4129a6c2f0292dd459f4ede0fa4b5c0d7c218023c87896772e6eRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl PYTORCH_WHL=torch-1.4.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.5.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:03 PM UTC
    sha256:5296a008e40412e7631f3b26226c0f659de3c227cb99314bdadb116713695936RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl PYTORCH_WHL=torch-1.4.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.5.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 6:50 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    PILLOW_VERSION=pillow<7
    04/27/2020 6:50 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHVISION_VERSION=v0.4.0
    04/27/2020 6:50 PM UTC
    sha256:39581c6148a220facee8b3c61a13852749277c7802abd4f115e1fc42e5bfac71RUN
    PYTORCH_URL=https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl PYTORCH_WHL=torch-1.4.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 6:50 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    PYTORCH_WHL=torch-1.2.0-cp36-cp36m-linux_aarch64.whl
    04/27/2020 6:49 PM UTC