Unverified Commit 064e886b authored by Min Xu's avatar Min Xu Committed by GitHub
Browse files

[chore] fix nightly download (#1019)


Co-authored-by: default avatarMin Xu <min.xu.public@gmail.com>
parent a342f349
......@@ -118,11 +118,11 @@ install_dep_pytorch_nightly: &install_dep_pytorch_nightly
# check if we have restored venv cache (/home/circleci/venv) correctly, if so, just skip
if [ -f /home/circleci/venv/check_version.py ]; then python /home/circleci/venv/check_version.py torch eq 1.12 && exit 0; fi
# start installing
pip install --progress-bar off --pre torch==1.12.0.dev20220424+cu113 torchvision==0.13.0.dev20220424+cu113 -f https://download.pytorch.org/whl/nightly/cu113/torch_nightly.html
pip install --progress-bar off --pre torch==1.13.0.dev20220625+cu113 torchvision==0.14.0.dev20220625+cu113 -f https://download.pytorch.org/whl/nightly/cu113/torch_nightly.html
pip install --progress-bar off -r requirements-dev.txt
pip install --progress-bar off -r requirements-benchmarks.txt
python -c 'import torch; print("Torch version:", torch.__version__)'
python -c 'import torch; assert torch.__version__.split(".")[:2] == ["1", "12"], "wrong torch version"'
python -c 'import torch; assert torch.__version__.split(".")[:2] == ["1", "13"], "wrong torch version"'
python -m torch.utils.collect_env
wget -O /home/circleci/venv/check_version.py https://raw.githubusercontent.com/min-xu-ai/check_verion/main/check_version.py
......
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