Add RTX 50-series CUDA support#50
Conversation
|
One heads-up for users who aren't on RTX 50-series: the |
Original PR (2026-04-29) pinned torch 2.7.1 / torchvision 0.22.1 — the latest CUDA-12.8 wheels available at the time. Bumping to current stable to keep the RTX 50 / Blackwell GPU support work current. CUDA 12.8 wheel index unchanged. Upstream main has moved 24 commits since the PR's merge base but none of them touched torch-requirements.txt, Dockerfile_*, build-*.sh, INSTALL.md, or segmentationRDS/segmentation.py — the rebase remains clean.
|
Quick refresh: bumped to current stable ( Checked the 24 commits on main since this PR's merge base — none of them touched Happy to make any further changes if useful. |
Summary
no kernel image is available for execution on the deviceContext
RTX 50-series / Blackwell GPUs require PyTorch binaries with CUDA 12.8+ support. The previous optional GPU requirements installed CUDA 12.4 wheels, which can expose CUDA but fail at runtime on RTX 50-series hardware. This addresses the failure reported downstream in alicevision/Meshroom#2950.
Tests
I did not build the Docker images locally.