Skip to content

GitLab

Unverified Commit ba00419a authored by moto's avatar moto Committed by GitHub
Browse files

Update package script to use UPLOAD_CHANNEL env (#962)

parent 6a0053e9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
packaging/pkg_helpers.bash
View file @ ba00419a
...@@ -171,8 +171,7 @@ setup_pip_pytorch_version() { ...@@ -171,8 +171,7 @@ setup_pip_pytorch_version() {
else else
pip_install "torch==$PYTORCH_VERSION$PYTORCH_VERSION_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/test/torch_test.html \ -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/torch_${UPLOAD_CHANNEL}.html"
-f https://download.pytorch.org/whl/nightly/torch_nightly.html
fi fi
} }
...@@ -185,7 +184,7 @@ setup_conda_pytorch_constraint() { ...@@ -185,7 +184,7 @@ setup_conda_pytorch_constraint() {
export CONDA_CHANNEL_FLAGS="-c pytorch-nightly" export CONDA_CHANNEL_FLAGS="-c pytorch-nightly"
export PYTORCH_VERSION="$(conda search --json 'pytorch[channel=pytorch-nightly]' | python -c "import sys, json, re; print(re.sub(r'\\+.*$', '', json.load(sys.stdin)['pytorch'][-1]['version']))")" export PYTORCH_VERSION="$(conda search --json 'pytorch[channel=pytorch-nightly]' | python -c "import sys, json, re; print(re.sub(r'\\+.*$', '', json.load(sys.stdin)['pytorch'][-1]['version']))")"
else else
export CONDA_CHANNEL_FLAGS="-c pytorch -c pytorch-test -c pytorch-nightly" export CONDA_CHANNEL_FLAGS="-c pytorch -c pytorch-${UPLOAD_CHANNEL}"
fi fi
if [[ "$CU_VERSION" == cpu ]]; then if [[ "$CU_VERSION" == cpu ]]; then
export CONDA_PYTORCH_BUILD_CONSTRAINT="- pytorch==$PYTORCH_VERSION${PYTORCH_VERSION_SUFFIX}" export CONDA_PYTORCH_BUILD_CONSTRAINT="- pytorch==$PYTORCH_VERSION${PYTORCH_VERSION_SUFFIX}"
......
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