set correct docker image

Signed-off-by: Edward Z. Yang <[email protected]>

Author: ezyang

.circleci/config.yml

@@ -195,11 +195,13 @@ workflows:
           name: binary_linux_wheel_py2.7_cu100
           python_version: "2.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_wheel:
           name: binary_linux_wheel_py2.7u_cu100
           python_version: "2.7"
           cu_version: "cu100"
           unicode_abi: "1"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_wheel:
           name: binary_linux_wheel_py2.7_cu101
           python_version: "2.7"
@@ -222,6 +224,7 @@ workflows:
           name: binary_linux_wheel_py3.5_cu100
           python_version: "3.5"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_wheel:
           name: binary_linux_wheel_py3.5_cu101
           python_version: "3.5"
@@ -239,6 +242,7 @@ workflows:
           name: binary_linux_wheel_py3.6_cu100
           python_version: "3.6"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_wheel:
           name: binary_linux_wheel_py3.6_cu101
           python_version: "3.6"
@@ -256,6 +260,7 @@ workflows:
           name: binary_linux_wheel_py3.7_cu100
           python_version: "3.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_wheel:
           name: binary_linux_wheel_py3.7_cu101
           python_version: "3.7"
@@ -294,6 +299,7 @@ workflows:
           name: binary_linux_conda_py2.7_cu100
           python_version: "2.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_conda:
           name: binary_linux_conda_py2.7_cu101
           python_version: "2.7"
@@ -311,6 +317,7 @@ workflows:
           name: binary_linux_conda_py3.5_cu100
           python_version: "3.5"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_conda:
           name: binary_linux_conda_py3.5_cu101
           python_version: "3.5"
@@ -328,6 +335,7 @@ workflows:
           name: binary_linux_conda_py3.6_cu100
           python_version: "3.6"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_conda:
           name: binary_linux_conda_py3.6_cu101
           python_version: "3.6"
@@ -345,6 +353,7 @@ workflows:
           name: binary_linux_conda_py3.7_cu100
           python_version: "3.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_linux_conda:
           name: binary_linux_conda_py3.7_cu101
           python_version: "3.7"
@@ -414,6 +423,7 @@ workflows:
           name: nightly_binary_linux_wheel_py2.7_cu100
           python_version: "2.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_wheel_upload:
           name: nightly_binary_linux_wheel_py2.7_cu100_upload
           context: org-member
@@ -425,6 +435,7 @@ workflows:
           python_version: "2.7"
           cu_version: "cu100"
           unicode_abi: "1"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_wheel_upload:
           name: nightly_binary_linux_wheel_py2.7u_cu100_upload
           context: org-member
@@ -477,6 +488,7 @@ workflows:
           name: nightly_binary_linux_wheel_py3.5_cu100
           python_version: "3.5"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_wheel_upload:
           name: nightly_binary_linux_wheel_py3.5_cu100_upload
           context: org-member
@@ -518,6 +530,7 @@ workflows:
           name: nightly_binary_linux_wheel_py3.6_cu100
           python_version: "3.6"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_wheel_upload:
           name: nightly_binary_linux_wheel_py3.6_cu100_upload
           context: org-member
@@ -559,6 +572,7 @@ workflows:
           name: nightly_binary_linux_wheel_py3.7_cu100
           python_version: "3.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_wheel_upload:
           name: nightly_binary_linux_wheel_py3.7_cu100_upload
           context: org-member
@@ -649,6 +663,7 @@ workflows:
           name: nightly_binary_linux_conda_py2.7_cu100
           python_version: "2.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_conda_upload:
           name: nightly_binary_linux_conda_py2.7_cu100_upload
           context: org-member
@@ -686,6 +701,7 @@ workflows:
           name: nightly_binary_linux_conda_py3.5_cu100
           python_version: "3.5"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_conda_upload:
           name: nightly_binary_linux_conda_py3.5_cu100_upload
           context: org-member
@@ -723,6 +739,7 @@ workflows:
           name: nightly_binary_linux_conda_py3.6_cu100
           python_version: "3.6"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_conda_upload:
           name: nightly_binary_linux_conda_py3.6_cu100_upload
           context: org-member
@@ -760,6 +777,7 @@ workflows:
           name: nightly_binary_linux_conda_py3.7_cu100
           python_version: "3.7"
           cu_version: "cu100"
+          wheel_docker_image: "soumith/manylinux-cuda100"
       - binary_conda_upload:
           name: nightly_binary_linux_conda_py3.7_cu100_upload
           context: org-member

.circleci/config.yml.in

@@ -176,6 +176,8 @@ jobs:
 {%-   endif %}
 {%-   if cu_version == "cu92" %}
           wheel_docker_image: "soumith/manylinux-cuda92"
+{%-   elif cu_version == "cu100" %}
+          wheel_docker_image: "soumith/manylinux-cuda100"
 {%-   endif %}
 
 {%-   if upload %}