Преглед на файлове

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