[BugFix] Fix builds (#849)

pytorch · Jul 3, 2024 · 53cdc61 · 53cdc61
1 parent a4ccecf
commit 53cdc61
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/nightly_build.yml b/.github/workflows/nightly_build.yml
@@ -42,7 +42,7 @@ jobs:
           ["3.12", "cp312-cp312"],
         ]
         cuda_support: [["", "cpu", "cpu"]]
-    container: pytorch/manylinux-cuda116
+    container: pytorch/manylinux-cuda121
     steps:
       - name: Checkout tensordict
         uses: actions/checkout@v2
@@ -51,6 +51,7 @@ jobs:
       - name: Install PyTorch nightly
         run: |
           export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
+          python3 -c """import sys;print(sys.version)"""
           python3 -mpip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda_support[1] }}
       - name: Build tensordict Nightly
         run: |