Skip to content

Update build_wheels.sh for Python 3.12 and torch 2.5.1#3

Open
tom-neara wants to merge 1 commit intoNeara-AI:mainfrom
tom-neara:update-py312-torch251
Open

Update build_wheels.sh for Python 3.12 and torch 2.5.1#3
tom-neara wants to merge 1 commit intoNeara-AI:mainfrom
tom-neara:update-py312-torch251

Conversation

@tom-neara
Copy link
Collaborator

Summary

  • Update Python 3.10 -> 3.12 and torch 2.4.1 -> 2.5.1 to match polez requirements
  • Pin setuptools<70 because mmcv's setup.py imports pkg_resources which was removed in setuptools 70+
  • Add --no-build-isolation so the build uses the venv's pinned setuptools instead of downloading the latest in an isolated env
  • Use cu124 PyTorch wheel index to match CUDA 12.4 toolkit

Test plan

  • Ran build_wheels.sh wheelhouse end-to-end on a machine with RTX 3090 / CUDA 12.4
  • All 5 wheels built successfully:
    • mmcv-2.1.0.20260302-0torch2.5.1numpy1.26.4-cp312-cp312-linux_x86_64.whl (194M, with CUDA ops)
    • mmdet-3.2.0.20260302-0torch2.5.1numpy1.26.4-py3-none-any.whl (6.8M)
    • mmdet3d-1.4.0.20260302-0torch2.5.1numpy1.26.4-py3-none-any.whl (3.6M)
    • mmengine-0.10.7.20260302-0torch2.5.1numpy1.26.4-py3-none-any.whl (1.7M)
    • torchsparse-1.4.0.20260302-0torch2.5.1numpy1.26.4-cp312-cp312-linux_x86_64.whl (32M)

🤖 Generated with Claude Code

- Python 3.10 -> 3.12, torch 2.4.1 -> 2.5.1 to match polez requirements
- Pin setuptools<70 because mmcv's setup.py imports pkg_resources which
  was removed in setuptools 70+
- Add --no-build-isolation so the build uses the pinned setuptools
- Use cu124 PyTorch index to match CUDA 12.4 toolkit

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
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