Eclectic-Sheep / sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric
https://eclecticsheep.ai
Apache License 2.0
305 stars 31 forks source link

Downgrade PyTorch to 2.1 due to NetworkError on windows #200

Closed belerico closed 8 months ago

belerico commented 8 months ago

Summary

torch==2.2 raises an torch.distributed.DistNetworkError: Unknown error error on Windows from the torch\distributed\rendezvous.py:174: DistNetworkError line. This PR reverts it back to 2.1

Type of Change

Please select the one relevant option below:

Checklist

Please confirm that the following tasks have been completed:

Thank you for your contribution! Once you have filled out this template, please ensure that you have assigned the appropriate reviewers and that all tests have passed.