Unverified Commit 4658f24d authored by Eli Uriegas's avatar Eli Uriegas Committed by GitHub
Browse files

Remove special versioning suffix rules for CUDA 10.2 (#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: default avatarEli Uriegas <eliuriegas@fb.com>

* also include windows cmath fix
Signed-off-by: default avatarEli Uriegas <eliuriegas@fb.com>

* remove comment
Signed-off-by: default avatarEli Uriegas <eliuriegas@fb.com>

* fix indent
Signed-off-by: default avatarEli Uriegas <eliuriegas@fb.com>

* actually add the script
Signed-off-by: default avatarEli Uriegas <eliuriegas@fb.com>
parent 34a8a37c
...@@ -35,10 +35,7 @@ setup_cuda() { ...@@ -35,10 +35,7 @@ setup_cuda() {
export WHEEL_DIR="" export WHEEL_DIR=""
# Wheel builds need suffixes (but not if they're on OS X, which never has suffix) # Wheel builds need suffixes (but not if they're on OS X, which never has suffix)
if [[ "$BUILD_TYPE" == "wheel" ]] && [[ "$(uname)" != Darwin ]]; then if [[ "$BUILD_TYPE" == "wheel" ]] && [[ "$(uname)" != Darwin ]]; then
# The default CUDA has no suffix export PYTORCH_VERSION_SUFFIX="+$CU_VERSION"
if [[ "$CU_VERSION" != "cu102" ]]; then
export PYTORCH_VERSION_SUFFIX="+$CU_VERSION"
fi
# Match the suffix scheme of pytorch, unless this package does not have # Match the suffix scheme of pytorch, unless this package does not have
# CUDA builds (in which case, use default) # CUDA builds (in which case, use default)
if [[ -z "$NO_CUDA_PACKAGE" ]]; then if [[ -z "$NO_CUDA_PACKAGE" ]]; then
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment