From 21fc4a74d8977ab222b86bb01d1e62cabc7d9981 Mon Sep 17 00:00:00 2001 From: Eli Uriegas <1700823+seemethere@users.noreply.github.com> Date: Wed, 24 Mar 2021 20:52:04 -0700 Subject: [PATCH] Remove special versioning suffix rules for CUDA 10.2 * 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 * also include windows cmath fix Signed-off-by: Eli Uriegas * remove comment Signed-off-by: Eli Uriegas * fix indent Signed-off-by: Eli Uriegas * actually add the script Signed-off-by: Eli Uriegas --- packaging/pkg_helpers.bash | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash index 826fb525e3a..48f7588594a 100644 --- a/packaging/pkg_helpers.bash +++ b/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