|
34 | 34 | base: "nvcr.io/nvidia/pytorch:24.08-py3" # CUDA 12.6 |
35 | 35 | - environment: PTLATEST+CUDA126 |
36 | 36 | pytorch: "-U torch torchvision --extra-index-url https://download.pytorch.org/whl/cu121" |
37 | | - base: "nvcr.io/nvidia/pytorch:24.09-py3" # CUDA 12.6 |
| 37 | + base: "nvcr.io/nvidia/pytorch:24.10-py3" # CUDA 12.6 |
38 | 38 | container: |
39 | 39 | image: ${{ matrix.base }} |
40 | 40 | options: "--gpus all" |
|
84 | 84 | if: github.repository == 'Project-MONAI/MONAI' |
85 | 85 | strategy: |
86 | 86 | matrix: |
87 | | - container: ["pytorch:23.08", "pytorch:24.08", "pytorch:24.09"] |
| 87 | + container: ["pytorch:23.08", "pytorch:24.08", "pytorch:24.10"] |
88 | 88 | container: |
89 | 89 | image: nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image |
90 | 90 | options: "--gpus all" |
@@ -133,7 +133,7 @@ jobs: |
133 | 133 | if: github.repository == 'Project-MONAI/MONAI' |
134 | 134 | strategy: |
135 | 135 | matrix: |
136 | | - container: ["pytorch:24.09"] |
| 136 | + container: ["pytorch:24.10"] |
137 | 137 | container: |
138 | 138 | image: nvcr.io/nvidia/${{ matrix.container }}-py3 # testing with the latest pytorch base image |
139 | 139 | options: "--gpus all" |
@@ -237,7 +237,7 @@ jobs: |
237 | 237 | if: github.repository == 'Project-MONAI/MONAI' |
238 | 238 | needs: cron-gpu # so that monai itself is verified first |
239 | 239 | container: |
240 | | - image: nvcr.io/nvidia/pytorch:24.09-py3 # testing with the latest pytorch base image |
| 240 | + image: nvcr.io/nvidia/pytorch:24.10-py3 # testing with the latest pytorch base image |
241 | 241 | options: "--gpus all --ipc=host" |
242 | 242 | runs-on: [self-hosted, linux, x64, integration] |
243 | 243 | steps: |
|
0 commit comments