From 333ba28c8a06ec309c3c9ff214e1d3e68ac36f01 Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Fri, 15 Nov 2019 19:18:49 +0100 Subject: [PATCH] Fix discrepancy in regenerate.py --- .circleci/regenerate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index e7d85d2f911..6925ffa6d45 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -62,9 +62,9 @@ def generate_base_workflow(base_workflow_name, python_version, cu_version, d["unicode_abi"] = '1' if cu_version == "cu92": - d["wheel_docker_image"] = "soumith/manylinux-cuda92" + d["wheel_docker_image"] = "pytorch/manylinux-cuda92" elif cu_version == "cu100": - d["wheel_docker_image"] = "soumith/manylinux-cuda100" + d["wheel_docker_image"] = "pytorch/manylinux-cuda100" if filter_branch is not None: d["filters"] = {"branches": {"only": filter_branch}}