Docs
circleci.com
Start Building for Free

Using the GPU execution environment

yesterday1 min read
Cloud
On This Page
  • Available resource classes
  • Linux GPU
  • Windows GPU
  • GPUs on server v2.x

You can run your jobs in the GPU execution environment, using either Windows or Linux virtual machines, for access to Nvidia GPUs for specialized workloads.

To use the Linux GPU execution environment, use the machine executor and specify a GPU-enabled image. For a full list of machine executor images see the CircleCI Developer Hub or the Configuration Reference.

version: 2.1

jobs:
  build:
    machine:
      image: ubuntu-2004-cuda-11.4:202110-01
    steps:
      - run: nvidia-smi

To use the Windows GPU execution environment, you can either choose to use the windows orb and specify the built-in GPU executor, or use the machine executor and specify a Windows GPU-enabled image. Refer to the Orb Registry page for full details, and the Developer Hub for full details of available machine executor images.

Available resource classes

Specify a resource class to fit your project and requirements. For further details on credit usage for these options, see the Resource Class pricing and plans page.

Linux GPU

version: 2.1

jobs:
  build:
    machine:
      image: ubuntu-2004-cuda-11.4:202110-01
    resource_class: gpu.nvidia.small
    steps:
      - run: nvidia-smi
ClassvCPUsRAMGPUsGPU modelGPU Memory (GiB)Disk Size (GiB)
gpu.nvidia.small4151Nvidia Tesla P48300
gpu.nvidia.medium8301Nvidia Tesla T416300
gpu.nvidia.large8301Nvidia Tesla V10016300

Note: These resources require review by our support team. Open a support ticket if you would like to request access.

Windows GPU

For Windows there is currently one resource class option. This will be used by default so you are not required to specify it in your configuration.

ClassvCPUsRAMGPUsGPU modelGPU Memory (GiB)Disk Size (GiB)
windows.gpu.nvidia.medium16601Nvidia Tesla T416200

Note: These resources require review by our support team. Open a support ticket if you would like to request access.

GPUs on server v2.x

If you are using CircleCI server v2.x, you can configure your VM service to use GPU-enabled machine executors. See Running GPU Executors in Server.


Help make this document better

This guide, as well as the rest of our docs, are open source and available on GitHub. We welcome your contributions.

Need support?

Our support engineers are available to help with service issues, billing, or account related questions, and can help troubleshoot build configurations. Contact our support engineers by opening a ticket.

You can also visit our support site to find support articles, community forums, and training resources.