heat
heat copied to clipboard
Proper setup of PyTorch versions in Conda based environments
Due Diligence
- General:
- [x] title of the PR is suitable to appear in the Release Notes
- Implementation:
- [ ] unit tests: all split configurations tested
- [ ] unit tests: multiple dtypes tested
- [ ] documentation updated where needed
Description
Issue/s resolved: #1491
Changes proposed:
- Restrict PyTorch versions to appropriate version range in Conda based environment.
Type of change
- bug fix
Memory requirements
Performance
Does this change modify the behaviour of other functions? If so, which?
no