From 69c767af3e5ac33e2786aa1e705cee6b3e0375ec Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Thu, 19 Dec 2024 10:20:20 +0000
Subject: [PATCH] Update

[ghstack-poisoned]
---
 .github/workflows/nightly_build.yml | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/nightly_build.yml b/.github/workflows/nightly_build.yml
index b18ee260f..dde4b4b61 100644
--- a/.github/workflows/nightly_build.yml
+++ b/.github/workflows/nightly_build.yml
@@ -32,7 +32,7 @@ jobs:
   build-wheel-linux:
     # Don't run on forked repos.
     if: github.repository_owner == 'pytorch'
-    runs-on: centos
+    runs-on: ubuntu-20.04
     strategy:
       matrix:
         python_version: [
@@ -42,10 +42,13 @@ jobs:
           ["3.12", "cp312-cp312"],
         ]
         cuda_support: [["", "cpu", "cpu"]]
-    container: pytorch/almalinux-builder
     steps:
       - name: Checkout tensordict
         uses: actions/checkout@v4
+      - name: Setup Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python_version[0] }}
       - name: Install PyTorch nightly
         run: |
           export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
@@ -76,7 +79,7 @@ jobs:
     # Don't run on forked repos.
     if: github.repository_owner == 'pytorch'
     needs: test-wheel-linux
-    runs-on: centos
+    runs-on: ubuntu-20.04
     strategy:
       matrix:
         python_version: [
@@ -86,10 +89,13 @@ jobs:
           ["3.12", "cp312-cp312"],
         ]
         cuda_support: [["", "cpu", "cpu"]]
-    container: pytorch/almalinux-builder
     steps:
       - name: Checkout tensordict
         uses: actions/checkout@v4
+      - name: Setup Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python_version[0] }}
       - name: Download built wheels
         uses: actions/download-artifact@v4
         with: