Skip to content

GitLab

Commit add2ecc7 authored by Soumith Chintala's avatar Soumith Chintala
Browse files

CUDA_SUFFIX -> PYTORCH_VERSION_SUFFIX

parent 32df87a8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
packaging/pkg_helpers.bash
View file @ add2ecc7
...@@ -184,7 +184,7 @@ setup_pip_pytorch_version() { ...@@ -184,7 +184,7 @@ setup_pip_pytorch_version() {
export PYTORCH_VERSION="$(pip show torch | grep ^Version: | sed 's/Version: *//')" export PYTORCH_VERSION="$(pip show torch | grep ^Version: | sed 's/Version: *//')"
fi fi
else else
pip_install "torch==$PYTORCH_VERSION$CUDA_SUFFIX" \ pip_install "torch==$PYTORCH_VERSION$PYTORCH_VERSION_SUFFIX" \
-f https://download.pytorch.org/whl/torch_stable.html \ -f https://download.pytorch.org/whl/torch_stable.html \
-f https://download.pytorch.org/whl/nightly/torch_nightly.html -f https://download.pytorch.org/whl/nightly/torch_nightly.html
fi fi
...@@ -193,7 +193,7 @@ setup_pip_pytorch_version() { ...@@ -193,7 +193,7 @@ setup_pip_pytorch_version() {
# Fill PYTORCH_VERSION with the latest conda nightly version, and # Fill PYTORCH_VERSION with the latest conda nightly version, and
# CONDA_CHANNEL_FLAGS with appropriate flags to retrieve these versions # CONDA_CHANNEL_FLAGS with appropriate flags to retrieve these versions
# #
# You MUST have populated CUDA_SUFFIX before hand. # You MUST have populated PYTORCH_VERSION_SUFFIX before hand.
setup_conda_pytorch_constraint() { setup_conda_pytorch_constraint() {
if [[ -z "$PYTORCH_VERSION" ]]; then if [[ -z "$PYTORCH_VERSION" ]]; then
export CONDA_CHANNEL_FLAGS="-c pytorch-nightly" export CONDA_CHANNEL_FLAGS="-c pytorch-nightly"
......
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