@@ -13,24 +13,28 @@ jobs:
1313 strategy :
1414 matrix :
1515 environment :
16- - " PT191+CUDA113"
1716 - " PT110+CUDA113"
18- - " PT113+CUDA113"
19- - " PTLATEST+CUDA121"
17+ - " PT113+CUDA118"
18+ - " PT210+CUDA121"
19+ - " PT240+CUDA126"
20+ - " PTLATEST+CUDA126"
2021 include :
2122 # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes
22- - environment : PT191+CUDA113
23- pytorch : " torch==1.9.1 torchvision==0.10.1 --extra-index-url https://download.pytorch.org/whl/cu113"
24- base : " nvcr.io/nvidia/pytorch:21.06-py3" # CUDA 11.3
2523 - environment : PT110+CUDA113
2624 pytorch : " torch==1.10.2 torchvision==0.11.3 --extra-index-url https://download.pytorch.org/whl/cu113"
2725 base : " nvcr.io/nvidia/pytorch:21.06-py3" # CUDA 11.3
28- - environment : PT113+CUDA113
29- pytorch : " torch==1.13.1 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu113"
30- base : " nvcr.io/nvidia/pytorch:21.06-py3" # CUDA 11.3
31- - environment : PTLATEST+CUDA121
32- pytorch : " -U torch torchvision --extra-index-url https://download.pytorch.org/whl/cu118"
33- base : " nvcr.io/nvidia/pytorch:23.08-py3" # CUDA 12.2
26+ - environment : PT113+CUDA118
27+ pytorch : " torch==1.13.1 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu121"
28+ base : " nvcr.io/nvidia/pytorch:22.10-py3" # CUDA 11.8
29+ - environment : PT210+CUDA121
30+ pytorch : " pytorch==2.1.0 torchvision==0.16.0 --extra-index-url https://download.pytorch.org/whl/cu121"
31+ base : " nvcr.io/nvidia/pytorch:23.08-py3" # CUDA 12.1
32+ - environment : PT240+CUDA126
33+ pytorch : " pytorch==2.4.0 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu121"
34+ base : " nvcr.io/nvidia/pytorch:24.08-py3" # CUDA 12.6
35+ - environment : PTLATEST+CUDA126
36+ pytorch : " -U torch torchvision --extra-index-url https://download.pytorch.org/whl/cu121"
37+ base : " nvcr.io/nvidia/pytorch:24.10-py3" # CUDA 12.6
3438 container :
3539 image : ${{ matrix.base }}
3640 options : " --gpus all"
5054 python -m pip install -r requirements-dev.txt
5155 python -m pip list
5256 - name : Run tests report coverage
57+ env :
58+ NGC_API_KEY : ${{ secrets.NGC_API_KEY }}
59+ NGC_ORG : ${{ secrets.NGC_ORG }}
60+ NGC_TEAM : ${{ secrets.NGC_TEAM }}
5361 run : |
5462 export LAUNCH_DELAY=$[ $RANDOM % 16 * 60 ]
5563 echo "Sleep $LAUNCH_DELAY"
7684 if : github.repository == 'Project-MONAI/MONAI'
7785 strategy :
7886 matrix :
79- container : ["pytorch:22.10 ", "pytorch:23 .08"]
87+ container : ["pytorch:23.08 ", "pytorch:24 .08", "pytorch:24.10 "]
8088 container :
8189 image : nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image
8290 options : " --gpus all"
@@ -94,6 +102,10 @@ jobs:
94102 python -m pip install -r requirements-dev.txt
95103 python -m pip list
96104 - name : Run tests report coverage
105+ env :
106+ NGC_API_KEY : ${{ secrets.NGC_API_KEY }}
107+ NGC_ORG : ${{ secrets.NGC_ORG }}
108+ NGC_TEAM : ${{ secrets.NGC_TEAM }}
97109 run : |
98110 export LAUNCH_DELAY=$[ $RANDOM % 16 * 60 ]
99111 echo "Sleep $LAUNCH_DELAY"
@@ -121,7 +133,7 @@ jobs:
121133 if : github.repository == 'Project-MONAI/MONAI'
122134 strategy :
123135 matrix :
124- container : ["pytorch:23.08 "]
136+ container : ["pytorch:24.10 "]
125137 container :
126138 image : nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image
127139 options : " --gpus all"
@@ -196,6 +208,10 @@ jobs:
196208 - name : Run tests report coverage
197209 # The docker image process has done the compilation.
198210 # BUILD_MONAI=1 is necessary for triggering the USE_COMPILED flag.
211+ env :
212+ NGC_API_KEY : ${{ secrets.NGC_API_KEY }}
213+ NGC_ORG : ${{ secrets.NGC_ORG }}
214+ NGC_TEAM : ${{ secrets.NGC_TEAM }}
199215 run : |
200216 cd /opt/monai
201217 nvidia-smi
@@ -221,7 +237,7 @@ jobs:
221237 if : github.repository == 'Project-MONAI/MONAI'
222238 needs : cron-gpu # so that monai itself is verified first
223239 container :
224- image : nvcr.io/nvidia/pytorch:23.08 -py3 # testing with the latest pytorch base image
240+ image : nvcr.io/nvidia/pytorch:24.10 -py3 # testing with the latest pytorch base image
225241 options : " --gpus all --ipc=host"
226242 runs-on : [self-hosted, linux, x64, integration]
227243 steps :
0 commit comments