Fix discrepancy in regenerate.py (#1583)

b05fc269 · Francisco Massa · GitHub · e4ba602e · b05fc269
Unverified Commit b05fc269 authored Nov 15, 2019 by Francisco Massa Committed by GitHub Nov 15, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

.circleci/regenerate.py .circleci/regenerate.py +2 -2

No files found.
--- 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}}