Remove special versioning suffix rules for CUDA 10.2 (pytorch#4065)

* 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]>
iutlu · Jun 15, 2021 · 4658f24 · 4658f24
1 parent 34a8a37
commit 4658f24
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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