diff --git a/.circleci/config.yml b/.circleci/config.yml index 09dd52d9..8c846628 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -337,6 +337,7 @@ workflows: python_version: '3.7' pytorch_version: 1.12.0 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 context: DOCKERHUB_TOKEN cu_version: cu116 name: linux_conda_py37_cu116_pyt1120 @@ -499,6 +500,7 @@ workflows: python_version: '3.8' pytorch_version: 1.12.0 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 context: DOCKERHUB_TOKEN cu_version: cu116 name: linux_conda_py38_cu116_pyt1120 @@ -661,6 +663,7 @@ workflows: python_version: '3.9' pytorch_version: 1.12.0 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 context: DOCKERHUB_TOKEN cu_version: cu116 name: linux_conda_py39_cu116_pyt1120 @@ -706,6 +709,7 @@ workflows: python_version: '3.10' pytorch_version: 1.12.0 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 context: DOCKERHUB_TOKEN cu_version: cu116 name: linux_conda_py310_cu116_pyt1120 diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index dcb27acb..6892afa4 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -33,11 +33,15 @@ CONDA_CUDA_VERSIONS = { def conda_docker_image_for_cuda(cuda_version): + if cuda_version in ("cu101", "cu102", "cu111"): + return None if cuda_version == "cu113": return "pytorch/conda-builder:cuda113" if cuda_version == "cu115": return "pytorch/conda-builder:cuda115" - return None + if cuda_version == "cu116": + return "pytorch/conda-builder:cuda116" + raise ValueError("Unknown cuda version") def pytorch_versions_for_python(python_version): diff --git a/packaging/linux_wheels/go.sh b/packaging/linux_wheels/go.sh index 359764c0..7f8eef58 100644 --- a/packaging/linux_wheels/go.sh +++ b/packaging/linux_wheels/go.sh @@ -8,3 +8,4 @@ sudo docker run --rm -v "$PWD/../../:/inside" pytorch/conda-cuda bash inside/packaging/linux_wheels/inside.sh sudo docker run --rm -v "$PWD/../../:/inside" -e SELECTED_CUDA=cu113 pytorch/conda-builder:cuda113 bash inside/packaging/linux_wheels/inside.sh sudo docker run --rm -v "$PWD/../../:/inside" -e SELECTED_CUDA=cu115 pytorch/conda-builder:cuda115 bash inside/packaging/linux_wheels/inside.sh +sudo docker run --rm -v "$PWD/../../:/inside" -e SELECTED_CUDA=cu116 pytorch/conda-builder:cuda116 bash inside/packaging/linux_wheels/inside.sh