Parcourir la source

Commenting out torch dependencies due to pip mishandling the cu113 flag related to proper CUDA version

NIX Worker il y a 2 ans
Parent
commit
5cef5492d4
1 fichiers modifiés avec 4 ajouts et 4 suppressions
  1. 4 4
      nnUNet/requirements.txt

+ 4 - 4
nnUNet/requirements.txt

@@ -14,7 +14,7 @@ matplotlib==3.4.3
 MedPy==0.4.0
 networkx==2.6.3
 nibabel==3.2.1
-nnunet==1.7.0
+#nnunet==1.7.0
 numpy==1.21.2
 packaging==21.0
 pandas==1.3.4
@@ -34,9 +34,9 @@ six==1.16.0
 sklearn==0.0
 threadpoolctl==3.0.0
 tifffile==2021.10.12
-torch==1.9.0+cu111
-torchaudio==0.9.1
-torchvision==0.10.0+cu111
+#torch==1.9.0+cu111
+#torchaudio==0.9.1
+#torchvision==0.10.0+cu111
 tqdm==4.62.3
 traceback2==1.4.0
 typing-extensions==3.10.0.2