This repository has been archived by the owner on Sep 13, 2024. It is now read-only.
Releases: peterjc123/pytorch-scripts
Releases · peterjc123/pytorch-scripts
PyTorch Windows version
Windows specific changes:
Bug fixes
- Errors in backward leads to deadlock
- Memory leak in multiprocessing using DataLoader
- An indention bug in torch.cuda
Features
- Add support for new CUDA and cuDNN versions
- Add support for Ninja and clcache when compiling
Known issues
- Some tests will fail
- There is no support for torch.distributed, nccl and magma
- No support for python versions other than 3.5 and 3.6
- Don't set num_worker to a number more than 1. ( 0 is safer ) Wrap your main code entry with an if statement like this.
if __name__ == '__main__':
The packages are accessible here.