Loosen torch pinning

By using ~=, we get broader compatiblity: on systems using `--index-url https://download.pytorch.org/whl/lts/1.8/cu102` or `--index-url https://download.pytorch.org/whl/lts/1.8/cu111` people will get 1.8.2 instead of 1.8.1, and will continue to get patches into the future so long as 1.8.x is maintained. See #1129 (comment) for the data that went into this.
ivadomed · May 26, 2022 · 2d7b598 · 2d7b598
1 parent a53e935
commit 2d7b598
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -21,6 +21,6 @@ tensorboard>=1.15.0
 tqdm>=4.30
 scipy
 torchio>=0.18.68
-torch==1.8.1
-torchvision==0.9.1
+torch~=1.8.1
+torchvision~=0.9.1
 wandb>=0.12.11