DirectorySecurity advisories
Sign in


pytorch-cuda12 logo


Last changed
Get started with latest for free
docker pull

Need access to a specific version? Contact us.

Sign in for updates

Get notified of upcoming product changes, critical vulnerability notifications and patches and more.

Sign in

A 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.

Download this Image

The image is available on

docker pull

Running pytorch-cuda12

PyTorch has some prerequisites which need to be configured in the environment prior to running with GPUs. For examples, please refer to

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:

docker run --rm -i -t \
    --privileged \
    --gpus all \

Testing GPU Access

If your environment has connected GPUs, you can check that PyTorch has access with the following:

docker run --rm -it --gpus all
Python 3.11.9 (main, Apr  2 2024, 15:40:32) [GCC 13.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
>>> torch.cuda.device_count()
>>> torch.cuda.get_device_name(0)
'Tesla V100-SXM2-16GB'

Testing PyTorch

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 script from the Chainguard Images repository, place it in a folder on your host machine, and run the script in a pytorch-cuda12 container from a volume.

mkdir pytorch-test &&\
 curl > pytorch-test/ &&\
 docker run --rm -it -v "$PWD/pytorch-test:/tmp/pytorch-test" --gpus all -c "python /tmp/pytorch-test/"

You may also consider running this quickstart script based on the official PyTorch quickstart tutorial using the same approach as above.

Using Helm charts

As a place to get started, you may also use this Helm chart to get PyTorch running

  helm install pytorch \
  --namespace pytorch-space --create-namespace  \
  --set image.registry="" \
  --set image.repository="chainguard/pytorch-cuda12" \
  --set image.tag=latest \
  --set containerSecurityContext.runAsUser=0 \
  --set containerSecurityContext.runAsNonRoot=false \
  --set containerSecurityContext.allowPrivilegeEscalation=true \
  --wait oci://

Chainguard Images contain software packages that are direct or transitive dependencies. The following licenses were found in the "latest" version of this image:

  • Apache-2.0

  • BSD-3-Clause

  • GCC-exception-3.1

  • GPL-2.0-or-later

  • GPL-3.0-or-later

  • IJG

  • LGPL-2.1-or-later

View more

For a complete list of licenses, please refer to this Image's SBOM.

Software license agreement



Chainguard Images

© 2024 Chainguard, Inc.