Update

[ghstack-poisoned]

Author: Vincent Moens

.github/unittest/linux/scripts/environment.yml

@@ -20,3 +20,4 @@ dependencies:
     - h5py
     - orjson
     - ninja
+    - numpy<2.0.0

.github/unittest/linux/scripts/setup_env.sh

@@ -8,6 +8,7 @@
 set -e
 set -v
 
+apt update -y && apt install git wget gcc -y
 
 this_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 # Avoid error: "fatal: unsafe repository"

.github/unittest/linux_torchrec/scripts/environment.yml

@@ -19,3 +19,4 @@ dependencies:
     - h5py
     - orjson
     - ninja
+    - numpy<2.0.0

.github/unittest/linux_torchrec/scripts/setup_env.sh

@@ -7,6 +7,8 @@
 
 set -e
 
+apt update -y && apt install git wget gcc -y
+
 this_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 # Avoid error: "fatal: unsafe repository"
 git config --global --add safe.directory '*'

.github/unittest/rl_linux_optdeps/scripts/environment.yml

@@ -17,3 +17,4 @@ dependencies:
     - scipy
     - orjson
     - ninja
+    - numpy<2.0.0

.github/unittest/rl_linux_optdeps/scripts/setup_env.sh

@@ -7,6 +7,8 @@
 
 set -e
 
+apt update -y && apt install git wget gcc -y
+
 this_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 # Avoid error: "fatal: unsafe repository"
 apt-get update && apt-get install -y git wget gcc g++

.github/workflows/docs.yml

@@ -23,7 +23,7 @@ jobs:
     strategy:
       matrix:
         python_version: ["3.10"]
-        cuda_arch_version: ["12.4"]
+        cuda_arch_version: ["12.8"]
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
       id-token: write

.github/workflows/test-linux.yml

@@ -23,14 +23,15 @@ jobs:
     strategy:
       matrix:
         python_version: ["3.10"]
-        cuda_arch_version: ["12.4"]
+        cuda_arch_version: ["12.8"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
       id-token: write
       contents: read
     with:
       runner: linux.g5.4xlarge.nvidia.gpu
+      docker-image: "nvidia/cuda:12.8.0-devel-ubuntu22.04"
       repository: pytorch/tensordict
       gpu-arch-type: cuda
       gpu-arch-version: ${{ matrix.cuda_arch_version }}
@@ -57,14 +58,15 @@ jobs:
   test-cpu:
     strategy:
       matrix:
-        python_version: ["3.9", "3.10", "3.11", "3.12"]
+        python_version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
       id-token: write
       contents: read
     with:
       runner: linux.12xlarge
+      docker-image: "nvidia/cuda:12.8.0-devel-ubuntu22.04"
       repository: pytorch/tensordict
       timeout: 90
       script: |
@@ -87,14 +89,15 @@ jobs:
     strategy:
       matrix:
         python_version: ["3.10"]
-        cuda_arch_version: ["12.4"]
+        cuda_arch_version: ["12.6"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
       id-token: write
       contents: read
     with:
       runner: linux.g5.4xlarge.nvidia.gpu
+      docker-image: "nvidia/cuda:12.6.0-devel-ubuntu22.04"
       repository: pytorch/tensordict
       gpu-arch-type: cuda
       gpu-arch-version: ${{ matrix.cuda_arch_version }}
@@ -122,14 +125,15 @@ jobs:
   test-stable-cpu:
     strategy:
       matrix:
-        python_version: ["3.9", "3.12"]
+        python_version: ["3.9", "3.13"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
       id-token: write
       contents: read
     with:
       runner: linux.12xlarge
+      docker-image: "nvidia/cuda:12.8.0-devel-ubuntu22.04"
       repository: pytorch/tensordict
       timeout: 90
       script: |

.github/workflows/test-rl-gpu.yml

@@ -23,14 +23,15 @@ jobs:
     strategy:
       matrix:
         python_version: ["3.10"]
-        cuda_arch_version: ["12.4"]
+        cuda_arch_version: ["12.8"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
       id-token: write
       contents: read
     with:
       runner: linux.g5.4xlarge.nvidia.gpu
+      docker-image: "nvidia/cuda:12.8.0-devel-ubuntu22.04"
       repository: pytorch/tensordict
       gpu-arch-type: cuda
       gpu-arch-version: ${{ matrix.cuda_arch_version }}