diff --git a/.github/workflows/test-linux.yml b/.github/workflows/test-linux.yml index 81121076a..c75c98d5f 100644 --- a/.github/workflows/test-linux.yml +++ b/.github/workflows/test-linux.yml @@ -25,7 +25,7 @@ jobs: python_version: ["3.10"] cuda_arch_version: ["12.1"] fail-fast: false - uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job.yml@pin-nvidia-container-toolkit with: runner: linux.g5.4xlarge.nvidia.gpu repository: pytorch/tensordict