Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion .github/workflows/pythonapp.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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()"
Expand Down
2 changes: 1 addition & 1 deletion docs/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -55,7 +55,7 @@ ignite =
torchvision =
torchvision
itk =
itk>=5.0
itk>=5.0, <=5.1.2
tqdm =
tqdm>=4.47.0
lmdb =
Expand Down