[v0.9.1] Remove special versioning suffix rules for CUDA 10.2 (#3603)

* packaging: Remove special versioning priveleges

Adds a version suffix for CUDA 10.2 for ease of installing when
installing from download.pytorch.org

Signed-off-by: Eli Uriegas <[email protected]>

* also include windows cmath fix

Signed-off-by: Eli Uriegas <[email protected]>

* remove comment

Signed-off-by: Eli Uriegas <[email protected]>

* fix indent

Signed-off-by: Eli Uriegas <[email protected]>

* actually add the script

Signed-off-by: Eli Uriegas <[email protected]>

Author: seemethere

.circleci/config.yml

@@ -294,6 +294,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Build conda packages
           no_output_timeout: 20m
@@ -322,6 +327,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Build wheel packages
           command: |
@@ -703,6 +713,11 @@ jobs:
       - checkout
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Generate cache key
           # This will refresh cache on Sundays, nightly build should generate new cache.
@@ -745,6 +760,11 @@ jobs:
       - checkout
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Generate cache key
           # This will refresh cache on Sundays, nightly build should generate new cache.
@@ -875,6 +895,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           command: |
             set -ex
@@ -889,6 +914,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           command: |
             set -ex

.circleci/config.yml.in

@@ -294,6 +294,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Build conda packages
           no_output_timeout: 20m
@@ -322,6 +327,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Build wheel packages
           command: |
@@ -703,6 +713,11 @@ jobs:
       - checkout
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Generate cache key
           # This will refresh cache on Sundays, nightly build should generate new cache.
@@ -745,6 +760,11 @@ jobs:
       - checkout
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           name: Generate cache key
           # This will refresh cache on Sundays, nightly build should generate new cache.
@@ -875,6 +895,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           command: |
             set -ex
@@ -889,6 +914,11 @@ jobs:
       - checkout_merge
       - designate_upload_channel
       - install_cuda_compatible_cmath
+      - run:
+          name: _HACK_ Install CUDA compatible cmath
+          no_output_timeout: 1m
+          command: |
+              powershell .circleci/scripts/vs_install_cmath.ps1
       - run:
           command: |
             set -ex

packaging/pkg_helpers.bash

@@ -35,10 +35,7 @@ setup_cuda() {
   export WHEEL_DIR=""
   # Wheel builds need suffixes (but not if they're on OS X, which never has suffix)
   if [[ "$BUILD_TYPE" == "wheel" ]] && [[ "$(uname)" != Darwin ]]; then
-    # The default CUDA has no suffix
-    if [[ "$CU_VERSION" != "cu102" ]]; then
-      export PYTORCH_VERSION_SUFFIX="+$CU_VERSION"
-    fi
+    export PYTORCH_VERSION_SUFFIX="+$CU_VERSION"
     # Match the suffix scheme of pytorch, unless this package does not have
     # CUDA builds (in which case, use default)
     if [[ -z "$NO_CUDA_PACKAGE" ]]; then