From 20a774e26623bb4a2fed5d0e5013c46d1026ea06 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Fri, 9 Dec 2022 07:31:39 -0800 Subject: [PATCH] builds for PyTorch 1.13.0 Summary: Add builds for current pytorch Reviewed By: shapovalov Differential Revision: D41841034 fbshipit-source-id: 4a58515bd5c83b26f14763a3cec9279e905391d2 --- .circleci/config.yml | 56 +++++++++++++++++++++++++++++++++++++++++ .circleci/regenerate.py | 3 +++ 2 files changed, 59 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index 585f09df..3b44218e 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -327,6 +327,20 @@ workflows: name: linux_conda_py37_cu116_pyt1121 python_version: '3.7' pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 + context: DOCKERHUB_TOKEN + cu_version: cu116 + name: linux_conda_py37_cu116_pyt1130 + python_version: '3.7' + pytorch_version: 1.13.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 + context: DOCKERHUB_TOKEN + cu_version: cu117 + name: linux_conda_py37_cu117_pyt1130 + python_version: '3.7' + pytorch_version: 1.13.0 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -474,6 +488,20 @@ workflows: name: linux_conda_py38_cu116_pyt1121 python_version: '3.8' pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 + context: DOCKERHUB_TOKEN + cu_version: cu116 + name: linux_conda_py38_cu116_pyt1130 + python_version: '3.8' + pytorch_version: 1.13.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 + context: DOCKERHUB_TOKEN + cu_version: cu117 + name: linux_conda_py38_cu117_pyt1130 + python_version: '3.8' + pytorch_version: 1.13.0 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -621,6 +649,20 @@ workflows: name: linux_conda_py39_cu116_pyt1121 python_version: '3.9' pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 + context: DOCKERHUB_TOKEN + cu_version: cu116 + name: linux_conda_py39_cu116_pyt1130 + python_version: '3.9' + pytorch_version: 1.13.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 + context: DOCKERHUB_TOKEN + cu_version: cu117 + name: linux_conda_py39_cu117_pyt1130 + python_version: '3.9' + pytorch_version: 1.13.0 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -687,6 +729,20 @@ workflows: name: linux_conda_py310_cu116_pyt1121 python_version: '3.10' pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 + context: DOCKERHUB_TOKEN + cu_version: cu116 + name: linux_conda_py310_cu116_pyt1130 + python_version: '3.10' + pytorch_version: 1.13.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 + context: DOCKERHUB_TOKEN + cu_version: cu117 + name: linux_conda_py310_cu117_pyt1130 + python_version: '3.10' + pytorch_version: 1.13.0 - binary_linux_conda_cuda: name: testrun_conda_cuda_py37_cu102_pyt190 context: DOCKERHUB_TOKEN diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 8fa8a446..4dd4b2c8 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -28,6 +28,7 @@ CONDA_CUDA_VERSIONS = { "1.11.0": ["cu102", "cu111", "cu113", "cu115"], "1.12.0": ["cu102", "cu113", "cu116"], "1.12.1": ["cu102", "cu113", "cu116"], + "1.13.0": ["cu116", "cu117"], } @@ -40,6 +41,8 @@ def conda_docker_image_for_cuda(cuda_version): return "pytorch/conda-builder:cuda115" if cuda_version == "cu116": return "pytorch/conda-builder:cuda116" + if cuda_version == "cu117": + return "pytorch/conda-builder:cuda117" raise ValueError("Unknown cuda version")