test-linux-envpool.yml

name: Envpool Tests on Linux

on:
  pull_request:
  push:
    branches:
      - nightly
      - main
      - release/*
  workflow_dispatch:

jobs:
  unittests:
    strategy:
      matrix:
        python_version: ["3.9"]
        cuda_arch_version: ["12.1"]
    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
    with:
      repository: pytorch/rl
      runner: "linux.g5.4xlarge.nvidia.gpu"
      # gpu-arch-type: cuda
      # gpu-arch-version: "11.7"
      # docker-image: "nvidia/cudagl:11.4.0-base"
      docker-image: "pytorch/manylinux-cuda117"
      timeout: 120
      script: |
        set -euo pipefail
        export PYTHON_VERSION="3.9"
        export CU_VERSION="12.1"
        export TAR_OPTIONS="--no-same-owner"
        export UPLOAD_CHANNEL="nightly"
        export TF_CPP_MIN_LOG_LEVEL=0

        nvidia-smi

        bash .github/unittest/linux_libs/scripts_envpool/setup_env.sh
        bash .github/unittest/linux_libs/scripts_envpool/install.sh
        bash .github/unittest/linux_libs/scripts_envpool/run_test.sh
        bash .github/unittest/linux_libs/scripts_envpool/post_process.sh