Repository navigation
chore(env): bump PyTorch to 2.11.0 and torchvision to 0.26.0 - #119
Closed
yuchenlichuck wants to merge 2 commits into
Closed
yuchenlichuck wants to merge 2 commits into
yuchenlichuck wants to merge 2 commits into
Conversation
Update the reproducible environment pins in requirements.txt from torch 2.10.0 / torchvision 0.25.0 to torch 2.11.0 / torchvision 0.26.0 (CUDA 12.8 wheels on Linux/Windows, native wheels on macOS), and update the matching Installation text in README.md. The library ranges in pyproject.toml (torch>=2.10,<2.12, torchvision>=0.25,<0.27) already admit 2.11.0 / 0.26.0, so they are unchanged. The PyTorch >= 2.10 floor for safe PSFNet checkpoint loading in deeplens/psfnetlens.py is unaffected. Verified: 672 tests pass on an RTX 6000 Ada (torch 2.11.0+cu128, torchvision 0.26.0+cu128, Python 3.12.12) with no warnings; ruff check / ruff format --check clean at 0.16.5. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
PyTorch 2.11 removed Volta (SM 7.0, e.g. V100) from the CUDA 12.8 and 12.9 binary builds (pytorch/pytorch#172598). requirements.txt pins the cu128 wheels, so tell V100 users to switch to the cu126 index, where torch 2.11.0+cu126 and torchvision 0.26.0+cu126 cp312 wheels are published. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
|
Verified against head
Suggestions (optional):
|
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Moves the reproducible environment from PyTorch 2.10.0 / torchvision 0.25.0 to PyTorch 2.11.0 / torchvision 0.26.0.
requirements.txt:torch==2.11.0+cu128/torchvision==0.26.0+cu128on Linux and Windows, nativetorch==2.11.0/torchvision==0.26.0wheels on macOS. Thecu128extra index is unchanged; 2.11.0 ships cp312 cu128 wheels formanylinux_2_28_x86_64,manylinux_2_28_aarch64andwin_amd64, and PyPI has themacosx_11_0_arm64wheels for both packages.README.md: Installation text updated to match.pyproject.toml: unchanged on purpose. The published-package rangestorch>=2.10,<2.12/torchvision>=0.25,<0.27already admit 2.11.0 / 0.26.0, following the split from 4aefe82 (library keeps ranges, env files carry the exact pins).deeplens/psfnetlens.py: the>= 2.10floor for safeweights_onlycheckpoint loading is unaffected and still correct.No source changes were needed; the library runs on 2.11 as-is.
Verification
Fresh Python 3.12.12 venv,
python -m pip install -r requirements.txt, thenpip install -e . --no-deps:torch 2.11.0+cu128,torchvision 0.26.0+cu128, CUDA 12.8 reported bytorch.version.cudapytest test/: 672 passed, 0 failed, 0 skipped, no warnings (RTX 6000 Ada, 88 s)0_hello_geolens.py,0_hello_defocuslens.py,0_hello_diffraclens.py,0_hello_hybridlens.pyall run to completion on GPUruff check .andruff format --check .clean at ruff 0.16.5🤖 Generated with Claude Code