Skip to content

Commit 95fe8a3

Browse files
Peter Romovfacebook-github-bot
authored andcommitted
Remove strict dependency on a specific PyTorch version (#135)
Summary: When building artifacts, PyTorch version is set strictly (`pytorch=xxx`) to the current version used in the build image. This forces reinstall of pytorch when installing csprng, although there is no actual reasons to use a specific version. Opacus is affected and blocked because of this problem: meta-pytorch/opacus#351 This PR changes deps in setup.py to just `pytorch` (without specifying bounds). Pull Request resolved: #135 Reviewed By: ffuuugor Differential Revision: D35408862 Pulled By: romovpa fbshipit-source-id: 30c18562e49ae439a31a005507b397713f40fd36
1 parent cd8f2f6 commit 95fe8a3

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

setup.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,12 +51,9 @@ def write_version_file():
5151
with open("README.md", "r") as fh:
5252
long_description = fh.read()
5353

54-
pytorch_dep = "torch"
55-
if os.getenv("PYTORCH_VERSION"):
56-
pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
5754

5855
requirements = [
59-
pytorch_dep,
56+
"torch",
6057
]
6158

6259

0 commit comments

Comments
 (0)