Update

[ghstack-poisoned]
pytorch · Nov 1, 2024 · 6e107f5 · 6e107f5
1 parent 5c781f1
commit 6e107f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test-linux.yml b/.github/workflows/test-linux.yml
@@ -54,7 +54,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
-      docker-image: pytorch/conda-builder:cuda12.1-996c40e85e70f1bc1adbd7772f1c9fe76bb00f63
+      docker-image: pytorch/conda-builder:cuda12.4-66a3c249ae3f9e136165ad17b90d619e713e9be1
       runner: linux.g5.4xlarge.nvidia.gpu
       repository: pytorch/tensordict
       gpu-arch-type: cuda
@@ -84,7 +84,7 @@ jobs:
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
-      docker-image: pytorch/conda-builder:cuda12.1-996c40e85e70f1bc1adbd7772f1c9fe76bb00f63
+      docker-image: pytorch/conda-builder:cuda12.4-66a3c249ae3f9e136165ad17b90d619e713e9be1
       runner: linux.g5.4xlarge.nvidia.gpu
       repository: pytorch/tensordict
       gpu-arch-type: cuda