Skip to content

Commit

Permalink
[spmd] pin pt version (#572)
Browse files Browse the repository at this point in the history
A recent change of c10d.ReduceOp crashes deepcopy of it, pin pt version
temporarily to fix CI.
see pytorch/pytorch#87303 (comment)
  • Loading branch information
wanchaol committed Oct 24, 2022
1 parent aa0be12 commit c74412b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions spmd/requirements.txt
Expand Up @@ -5,10 +5,10 @@ pyyaml

--pre
--extra-index-url https://download.pytorch.org/whl/nightly/cpu
torch >= 1.14.0.dev0; sys_platform == "darwin"
torch >= 1.14.0.dev0, <= 1.14.0dev20221021; sys_platform == "darwin"

--pre
--extra-index-url https://download.pytorch.org/whl/nightly/cu113
--extra-index-url https://download.pytorch.org/whl/nightly/cu116
torch >= 1.14.0.dev0; sys_platform == "linux"
torch >= 1.14.0.dev0, <= 1.14.0dev20221021; sys_platform == "linux"

0 comments on commit c74412b

Please sign in to comment.