Skip to content

Commit 4658f24

Browse files
authored
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: 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]>
1 parent 34a8a37 commit 4658f24

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

packaging/pkg_helpers.bash

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,7 @@ setup_cuda() {
3535
export WHEEL_DIR=""
3636
# Wheel builds need suffixes (but not if they're on OS X, which never has suffix)
3737
if [[ "$BUILD_TYPE" == "wheel" ]] && [[ "$(uname)" != Darwin ]]; then
38-
# The default CUDA has no suffix
39-
if [[ "$CU_VERSION" != "cu102" ]]; then
40-
export PYTORCH_VERSION_SUFFIX="+$CU_VERSION"
41-
fi
38+
export PYTORCH_VERSION_SUFFIX="+$CU_VERSION"
4239
# Match the suffix scheme of pytorch, unless this package does not have
4340
# CUDA builds (in which case, use default)
4441
if [[ -z "$NO_CUDA_PACKAGE" ]]; then

0 commit comments

Comments
 (0)