Skip to content

Enable PyTorch CUDA 12.8 support for RTX 5070 Ti#59

Open
AnthonyVelasco wants to merge 1 commit intoCBBIO:test-upload-v2from
AnthonyVelasco:antonio-torch-update
Open

Enable PyTorch CUDA 12.8 support for RTX 5070 Ti#59
AnthonyVelasco wants to merge 1 commit intoCBBIO:test-upload-v2from
AnthonyVelasco:antonio-torch-update

Conversation

@AnthonyVelasco
Copy link
Copy Markdown

This PR updates the PyTorch stack to use CUDA 12.8, which enables GPU execution
on newer architectures (e.g. RTX 5070 Ti / sm_120).

Summary:

  • Switched PyTorch source to cu128
  • Updated torch / torchvision versions accordingly
  • Verified successful GPU execution (seconds instead of minutes)

This is proposed from a fork because I do not have push access to the main repo.

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