Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit cfbf09d9 authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Update 'torch' extra dependency specifiers.

- Drop support for Torch 1.10-1.12, which complexify rules for functorch.
- Make "torch", "torch1" and "torch2" specifiers deal merely with Torch,
  leaving Opacus to the "dp" specifier, as should be.
- Have functorch be installed together with torch, removing some burden
  from our specifier, and enabling support for the new Torch 2.1.
parent edd2d777
No related branches found
No related tags found
No related merge requests found
Pipeline #866040 passed
......@@ -73,17 +73,12 @@ tensorflow = [
"tensorflow ~= 2.5",
]
torch = [ # generic requirements for Torch
"functorch >= 1.10, < 3.0", # note: unused with torch >=2.0
"torch >= 1.10, < 3.0",
"torch >= 1.13, < 3.0",
]
torch1 = [ # Torch 1.13 (latest pre-2.0 version) + Opacus
"functorch ~= 1.13.0",
"opacus ~= 1.3.0",
torch1 = [ # Torch 1.13 (latest pre-2.0 version)
"torch ~= 1.13.0",
]
torch2 = [ # Torch 2.X version + Opacus
"functorch ~= 2.0", # unused, but useful to linters and pip
"opacus ~= 1.4",
torch2 = [ # Torch 2.X version
"torch ~= 2.0",
]
websockets = [
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment