Переглянути джерело

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