fixes itk dep. version (#1944)

Signed-off-by: Wenqi Li <[email protected]>

Author: wyli

.github/workflows/pythonapp.yml

@@ -281,12 +281,19 @@ jobs:
         python -m pip install --upgrade pip wheel
         python -m pip install ${{ matrix.pytorch }}
         python -m pip install -r requirements-dev.txt
+        python -m pip list
     - name: Run quick tests (GPU)
       run: |
-        python -m pip list
         nvidia-smi
+        export LAUNCH_DELAY=$(( RANDOM % 30 * 5 ))
+        echo "Sleep $LAUNCH_DELAY"
+        sleep $LAUNCH_DELAY
         export CUDA_VISIBLE_DEVICES=$(coverage run -m tests.utils)
         echo $CUDA_VISIBLE_DEVICES
+        stop_time=$((LAUNCH_DELAY + $(date +%s)))
+        while [ $(date +%s) -lt $stop_time ]; do
+          python -c 'import torch; torch.rand(5, 3, device=torch.device("cuda:0"))';
+        done
         python -c "import torch; print(torch.__version__); print('{} of GPUs available'.format(torch.cuda.device_count()))"
         python -c 'import torch; print(torch.rand(5, 3, device=torch.device("cuda:0")))'
         python -c "import monai; monai.config.print_config()"

docs/requirements.txt

@@ -2,7 +2,7 @@
 torch>=1.5
 pytorch-ignite==0.4.4
 numpy>=1.17
-itk>=5.0
+itk>=5.0, <=5.1.2
 nibabel
 cucim==0.18.2
 openslide-python==1.1.2

requirements-dev.txt

@@ -3,7 +3,7 @@
 pytorch-ignite==0.4.4
 gdown>=3.6.4
 scipy
-itk>=5.0
+itk>=5.0, <=5.1.2
 nibabel
 pillow
 tensorboard

setup.cfg

@@ -36,7 +36,7 @@ all =
     pytorch-ignite==0.4.4
     gdown>=3.6.4
     torchvision
-    itk>=5.0
+    itk>=5.0, <=5.1.2
     tqdm>=4.47.0
     cucim==0.18.2
     openslide-python==1.1.2
@@ -55,7 +55,7 @@ ignite =
 torchvision =
     torchvision
 itk =
-    itk>=5.0
+    itk>=5.0, <=5.1.2
 tqdm =
     tqdm>=4.47.0
 lmdb =