diff --git a/examples/pytorch/BertNewsClassification/conda.yaml b/examples/pytorch/BertNewsClassification/conda.yaml index 77f6412ab39c0..af74b75692834 100644 --- a/examples/pytorch/BertNewsClassification/conda.yaml +++ b/examples/pytorch/BertNewsClassification/conda.yaml @@ -14,3 +14,10 @@ dependencies: - torch>=1.9.0 - torchtext==0.10.0 - pytorch-lightning==1.5.1 + # torch <= 1.10.1 doesn't work with setuptools >= 59.6.0. + # See the following PRs for details: + # https://github.com/pytorch/pytorch/pull/69904 + # https://github.com/pytorch/pytorch/pull/69947 + # + # TODO: Remove this requirement once torch > 1.10.1 is released. + - setuptools<59.6.0 diff --git a/examples/pytorch/IterativePruning/conda.yaml b/examples/pytorch/IterativePruning/conda.yaml index f80895027a185..1a32768a46abb 100644 --- a/examples/pytorch/IterativePruning/conda.yaml +++ b/examples/pytorch/IterativePruning/conda.yaml @@ -11,3 +11,10 @@ dependencies: - ax-platform - prettytable - torch>=1.9.0 + # torch <= 1.10.1 doesn't work with setuptools >= 59.6.0. + # See the following PRs for details: + # https://github.com/pytorch/pytorch/pull/69904 + # https://github.com/pytorch/pytorch/pull/69947 + # + # TODO: Remove this requirement once torch > 1.10.1 is released. + - setuptools<59.6.0 diff --git a/examples/pytorch/MNIST/conda.yaml b/examples/pytorch/MNIST/conda.yaml index 5246dd669579e..d16853c2c6d76 100644 --- a/examples/pytorch/MNIST/conda.yaml +++ b/examples/pytorch/MNIST/conda.yaml @@ -8,3 +8,10 @@ dependencies: - torchvision>=0.9.1 - torch>=1.9.0 - pytorch-lightning==1.5.1 + # torch <= 1.10.1 doesn't work with setuptools >= 59.6.0. + # See the following PRs for details: + # https://github.com/pytorch/pytorch/pull/69904 + # https://github.com/pytorch/pytorch/pull/69947 + # + # TODO: Remove this requirement once torch > 1.10.1 is released. + - setuptools<59.6.0