Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "6935a818eca9246734b6ee7504d6e0fa5fabe3b5"
Unverified Commit 7aea80c9 authored by Eli Uriegas's avatar Eli Uriegas Committed by GitHub
Browse files

Add test channels for pytorch version functions (#2208)

parent 40ad5572
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
packaging/pkg_helpers.bash
View file @ 7aea80c9
...@@ -214,6 +214,7 @@ setup_pip_pytorch_version() { ...@@ -214,6 +214,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/nightly/torch_nightly.html -f https://download.pytorch.org/whl/nightly/torch_nightly.html
fi fi
} }
...@@ -239,7 +240,7 @@ setup_conda_pytorch_constraint() { ...@@ -239,7 +240,7 @@ setup_conda_pytorch_constraint() {
exit 1 exit 1
fi fi
else else
export CONDA_CHANNEL_FLAGS="-c pytorch -c pytorch-nightly" export CONDA_CHANNEL_FLAGS="-c pytorch -c pytorch-nightly -c pytorch-test"
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