Restrict PyTorch version to <2.3.0 to resolve import error #577
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.
Fixes #576
This addresses the import error encountered with PyTorch 2.3.0 as detailed in issue #576. The error 'cannot import name '_refresh_per_optimizer_state' from 'torch.cuda.amp.grad_scaler' is resolved by pinning the PyTorch version to 2.2.2 in setup.cfg.
Small change:
setup.cfg
to specifytorch==2.2.2
underinstall_requires
.