From c63980d0d7e5c3962e4c9c4112f9906937eac06b Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Mon, 28 Oct 2024 14:04:17 -0700 Subject: [PATCH] Update container to 24.03 --- .github/workflows/nv-a6000.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml index b721357795b4..99adbca7c64c 100644 --- a/.github/workflows/nv-a6000.yml +++ b/.github/workflows/nv-a6000.yml @@ -23,7 +23,7 @@ jobs: unit-tests: runs-on: [self-hosted, nvidia, a6000] container: - image: nvcr.io/nvidia/pytorch:23.09-py3 + image: nvcr.io/nvidia/pytorch:24.03-py3 ports: - 80 options: --gpus all --shm-size "8G"