pytorch / pytorch/pytorch

[CI][Windows][Infrastructure] Support for RTX GPU workflows on Windows

Open
#170,753 4 comments 0 reactions 0 assignees View on GitHub
module: ci module: windows triaged
Dominant language
Python
Stars
103k
Forks
29.5k
PR merge metrics
PR metrics pending

Description

Description: This issue tracks the effort to enable Windows CI workflows with support for RTX series GPUs.

Motivation Currently, PyTorch Windows CI infrastructure has no coverage for RTX specific workloads. Adding this support will allow us to:

Validate PyTorch functionality and performance on consumer-grade NVIDIA hardware on Windows.

Catch regressions specific to the RTX architecture earlier in the development cycle.

Better support the growing community of developers using PyTorch on Windows with RTX GPUs.

Proposed Changes

Provision Windows runners equipped with RTX GPUs into the CI pipeline.

Update workflow configurations to target these new runners for relevant test suites.

Verify stability and performance baselines.

Status

[✅] Infrastructure provisioning

[✅ ] CI Workflow integration

[✅ ] Validation and testing

cc @peterjc123 @mszhanyi @skyline75489 @nbcsm @iremyux @Blackhex @seemethere @malfet @pytorch/pytorch-dev-infra

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.