Skip to content

pin images to cudnn cu128. this shold break ci because of the vllm de…#2451

Open
Jackmin801 wants to merge 1 commit intomainfrom
unify-ci-and-build-image
Open

pin images to cudnn cu128. this shold break ci because of the vllm de…#2451
Jackmin801 wants to merge 1 commit intomainfrom
unify-ci-and-build-image

Conversation

@Jackmin801
Copy link
Copy Markdown
Member

@Jackmin801 Jackmin801 commented May 8, 2026

…fault being cu13


Note

Medium Risk
Changes the container runtime for GPU unit/integration and nightly workflows, which can break CI due to different preinstalled libraries (e.g., PyTorch) and CUDA/cuDNN compatibility.

Overview
Updates the GPU CI environment by replacing the pytorch/pytorch:2.10.0-cuda13.0-cudnn9-devel container with nvidia/cuda:12.8.1-cudnn-devel-ubuntu22.04 for GPU unit tests, GPU integration tests, and nightly experiments.

This effectively pins workflows to CUDA 12.8 + cuDNN and removes reliance on the PyTorch-provided image, changing what system/ML dependencies are present at runtime.

Reviewed by Cursor Bugbot for commit 987f9f0. Bugbot is set up for automated code reviews on this repo. Configure here.

@Jackmin801 Jackmin801 marked this pull request as draft May 8, 2026 23:08
@Jackmin801 Jackmin801 force-pushed the unify-ci-and-build-image branch from 987f9f0 to 0ef6200 Compare May 9, 2026 00:32
@Jackmin801 Jackmin801 marked this pull request as ready for review May 9, 2026 00:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant