Skip to content

ENH: auto-detect PyTorch CUDA version for virtual environment setup - #4766

Merged
qinxuye merged 2 commits into
xorbitsai:mainfrom
qinxuye:fix/virtual-env
Apr 3, 2026
Merged

ENH: auto-detect PyTorch CUDA version for virtual environment setup#4766
qinxuye merged 2 commits into
xorbitsai:mainfrom
qinxuye:fix/virtual-env

Conversation

@qinxuye

@qinxuye qinxuye commented Apr 3, 2026

Copy link
Copy Markdown
Contributor
  • Add CUDA version detection from system PyTorch packages
  • Auto-configure PyTorch wheel URLs based on detected CUDA version
  • Support CUDA versions cu128, cu129, cu130
  • Preserve CUDA suffixes when resolving system package requirements

Fixes #4625

Xoscar need to be >=0.9.5

@XprobeBot XprobeBot added enhancement New feature or request gpu labels Apr 3, 2026
@XprobeBot XprobeBot added this to the v2.x milestone Apr 3, 2026
@qinxuye
qinxuye requested a review from amumu96 April 3, 2026 14:35

@amumu96 amumu96 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

qinxuye added 2 commits April 3, 2026 23:16
- Add CUDA version detection from system PyTorch packages
- Auto-configure PyTorch wheel URLs based on detected CUDA version
- Support CUDA versions cu128, cu129, cu130
- Preserve CUDA suffixes when resolving system package requirements
- Keep #system_torch# and other markers unresolved until installation phase
- Auto-configure PyTorch wheel URLs based on detected CUDA version from system
- Remove unused resolve_system_requirements_with_cuda function
- Fix test failure when torchaudio is not installed in test environment
@qinxuye
qinxuye merged commit b2f52ea into xorbitsai:main Apr 3, 2026
12 of 14 checks passed
@qinxuye
qinxuye deleted the fix/virtual-env branch April 3, 2026 16:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request gpu

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Failed to start Qwen3-ASR-0.6B in xinference docker with returned non-zero exit status 1.

3 participants