@@ -26,13 +26,13 @@ jobs:
2626 tests-cpu :
2727 strategy :
2828 matrix :
29- python_version : ["3.9 ", "3.10 ", "3.11 ", "3.12 "]
29+ python_version : ["3.10 ", "3.11 ", "3.12 ", "3.13", "3.14 "]
3030 fail-fast : false
3131 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
3232 with :
3333 runner : linux.12xlarge
3434 repository : pytorch/rl
35- docker-image : " nvidia/cuda:12.2.0- devel-ubuntu22 .04"
35+ docker-image : " nvidia/cuda:13.0.2-cudnn- devel-ubuntu24 .04"
3636 timeout : 90
3737 script : |
3838 if [[ "${{ github.ref }}" =~ release/* ]]; then
@@ -56,14 +56,14 @@ jobs:
5656 tests-gpu :
5757 strategy :
5858 matrix :
59- python_version : ["3.11 "]
60- cuda_arch_version : ["12.8 "]
59+ python_version : ["3.12 "]
60+ cuda_arch_version : ["13.0 "]
6161 fail-fast : false
6262 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
6363 with :
6464 runner : linux.g5.4xlarge.nvidia.gpu
6565 repository : pytorch/rl
66- docker-image : " nvidia/cuda:12.4.0- devel-ubuntu22 .04"
66+ docker-image : " nvidia/cuda:13.0.2-cudnn- devel-ubuntu24 .04"
6767 gpu-arch-type : cuda
6868 gpu-arch-version : ${{ matrix.cuda_arch_version }}
6969 timeout : 90
@@ -128,14 +128,14 @@ jobs:
128128 tests-optdeps :
129129 strategy :
130130 matrix :
131- python_version : ["3.11 "]
132- cuda_arch_version : ["12.8 "]
131+ python_version : ["3.12 "]
132+ cuda_arch_version : ["13.0 "]
133133 fail-fast : false
134134 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
135135 with :
136136 runner : linux.g5.4xlarge.nvidia.gpu
137137 repository : pytorch/rl
138- docker-image : " nvidia/cuda:12.4.0- devel-ubuntu22 .04"
138+ docker-image : " nvidia/cuda:13.0.2-cudnn- devel-ubuntu24 .04"
139139 gpu-arch-type : cuda
140140 gpu-arch-version : ${{ matrix.cuda_arch_version }}
141141 timeout : 90
@@ -163,14 +163,14 @@ jobs:
163163 tests-stable-gpu :
164164 strategy :
165165 matrix :
166- python_version : ["3.10 "] # "3.9", "3.10", "3.11"
167- cuda_arch_version : ["11.8 "] # "11.6", "11.7"
166+ python_version : ["3.12 "] # "3.9", "3.10", "3.11"
167+ cuda_arch_version : ["13.0 "] # "11.6", "11.7"
168168 fail-fast : false
169169 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
170170 with :
171171 runner : linux.g5.4xlarge.nvidia.gpu
172172 repository : pytorch/rl
173- docker-image : " nvidia/cuda:12.4.0- devel-ubuntu22 .04"
173+ docker-image : " nvidia/cuda:13.0.2-cudnn- devel-ubuntu24 .04"
174174 gpu-arch-type : cuda
175175 gpu-arch-version : ${{ matrix.cuda_arch_version }}
176176 timeout : 90
0 commit comments