Last changed
docker pull cgr.dev/chainguard/pytorch
Need access to a specific tag? Contact us.
Integrate Chainguard into your developer workflows, manage your image versions to stay free of CVEs, and view critical SBOM and provenance details.
Sign upA minimal, wolfi-based image for pytorch, a Python package that provides two high-level features: Tensor computation with strong GPU acceleration and Deep neural networks built on a tape-based autograd system.
Chainguard Containers are regularly-updated, secure-by-default container images.
For those with access, this container image is available on cgr.dev
:
Be sure to replace the ORGANIZATION
placeholder with the name used for your organization's private repository within the Chainguard Registry.
Chainguard’s PyTorch image is similar to the pytorch/pytorch image, with several key differences:
pytorch/pytorch
images are based on Ubuntu 22.04nonroot
user with home in /home/nonroot
directory, while the pytorch/pytorch
images are running as root userpytorch/pytorch
uses Bash as entrypoint
(NOTE: Chainguard also provides a -dev
images that have a full shell and entrypoint can be set to /bin/sh
or /bin/bash
for such images)If you're running an older version of CUDA not supported by the container, you have the option to install CUDA compatibility packages.
First, install the compatibility package for your specific host OS using the NVIDIA package repository. Make sure to install the package specific to your current version of CUDA.
Once the compatibility package has been installed, you can run the container in compatibility mode:
PyTorch has some prerequisites which need to be configured in the environment prior to running with GPUs. For examples, please refer to TESTING.md.
Additionally, please refer to the upstream documentation for more information on configuring and using PyTorch.
Assuming the environment prerequisites have been met, below demonstrates how to launch the container:
If your environment has connected GPUs, you can check that PyTorch has access with the following:
As a quick intro, we will use PyTorch to create a very simple deep learning model with two linear layers and an activation function. We’ll create an instance of it and ask it to report on its parameters. Running the below will fetch a model_builder.py script from the Chainguard Images repository, place it in a folder on your host machine, and run the script in a pytorch container from a volume.
You may also consider running this quickstart script based on the official PyTorch quickstart tutorial using the same approach as above.
As a place to get started, you may also use this Helm chart to get PyTorch running
Chainguard's free tier of Starter container images are built with Wolfi, our minimal Linux undistro.
All other Chainguard Containers are built with Chainguard OS, Chainguard's minimal Linux operating system designed to produce container images that meet the requirements of a more secure software supply chain.
The main features of Chainguard Containers include:
For cases where you need container images with shells and package managers to build or debug, most Chainguard Containers come paired with a development, or -dev
, variant.
In all other cases, including Chainguard Containers tagged as :latest
or with a specific version number, the container images include only an open-source application and its runtime dependencies. These minimal container images typically do not contain a shell or package manager.
Although the -dev
container image variants have similar security features as their more minimal versions, they include additional software that is typically not necessary in production environments. We recommend using multi-stage builds to copy artifacts from the -dev
variant into a more minimal production image.
To improve security, Chainguard Containers include only essential dependencies. Need more packages? Chainguard customers can use Custom Assembly to add packages, either through the Console, chainctl
, or API.
To use Custom Assembly in the Chainguard Console: navigate to the image you'd like to customize in your Organization's list of images, and click on the Customize image button at the top of the page.
Refer to our Chainguard Containers documentation on Chainguard Academy. Chainguard also offers VMs and Libraries — contact us for access.
This software listing is packaged by Chainguard. The trademarks set forth in this offering are owned by their respective companies, and use of them does not imply any affiliation, sponsorship, or endorsement by such companies.
Chainguard container images contain software packages that are direct or transitive dependencies. The following licenses were found in the "latest" tag of this image:
Apache-2.0
BSD-1-Clause
BSD-2-Clause
BSD-3-Clause
BSD-4-Clause-UC
CC-PDDC
FTL
For a complete list of licenses, please refer to this Image's SBOM.
Software license agreementA FIPS validated version of this image is available for FedRAMP compliance. STIG is included with FIPS image.