Unverified Commit eb28c650 authored by Tim Moon's avatar Tim Moon Committed by GitHub
Browse files

Update build test CUDA version to 12.1 (#1517)


Signed-off-by: default avatarTim Moon <tmoon@nvidia.com>
parent 2834e4ad
...@@ -12,7 +12,7 @@ jobs: ...@@ -12,7 +12,7 @@ jobs:
name: 'Core' name: 'Core'
runs-on: ubuntu-latest runs-on: ubuntu-latest
container: container:
image: nvcr.io/nvidia/cuda:12.0.0-devel-ubuntu22.04 image: nvcr.io/nvidia/cuda:12.1.0-devel-ubuntu22.04
options: --user root options: --user root
steps: steps:
- name: 'Dependencies' - name: 'Dependencies'
...@@ -35,7 +35,7 @@ jobs: ...@@ -35,7 +35,7 @@ jobs:
name: 'PyTorch' name: 'PyTorch'
runs-on: ubuntu-latest runs-on: ubuntu-latest
container: container:
image: nvcr.io/nvidia/cuda:12.5.0-devel-ubuntu22.04 image: nvcr.io/nvidia/cuda:12.8.0-devel-ubuntu22.04
options: --user root options: --user root
steps: steps:
- name: 'Dependencies' - name: 'Dependencies'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment