瀏覽代碼

Downgrading torch and torchvision to match installed version under studen

NIX Worker 2 年之前
父節點
當前提交
6ec84bb34e
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      nnUNet/requirements.txt

+ 2 - 2
nnUNet/requirements.txt

@@ -34,9 +34,9 @@ six==1.16.0
 sklearn==0.0
 threadpoolctl==3.0.0
 tifffile==2021.10.12
-torch==1.9.1+cu111
+torch==1.9.0+cu111
 torchaudio==0.9.1
-torchvision==0.10.1+cu111
+torchvision==0.10.0+cu111
 tqdm==4.62.3
 traceback2==1.4.0
 typing-extensions==3.10.0.2