Install torchvision from master

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/20836

Differential Revision: D15464705

Pulled By: bddppq

fbshipit-source-id: abe2ac2de2bf4c8d07334e6b2565c738c40428ae
This commit is contained in:
Junjie Bai
2019-05-23 02:10:47 -07:00
committed by Facebook Github Bot
parent d35a587958
commit 90182a7332
2 changed files with 15 additions and 2 deletions

View File

@ -120,6 +120,5 @@ pip install --user pytest-sugar
# torchvision tests #
#####################
if [[ "$BUILD_ENVIRONMENT" == *onnx* ]]; then
pip install --user torchvision
"$ROOT_DIR/scripts/onnx/test.sh"
fi

View File

@ -23,7 +23,21 @@ do
done
set -- "${UNKNOWN[@]}" # leave UNKNOWN
pip install pytest scipy torchvision hypothesis
pip install pytest scipy hypothesis
install_torchvision() {
echo "Installing torchvision at branch master"
rm -rf vision
# TODO: This git clone is bad, it means pushes to torchvision can break
# PyTorch CI
git clone https://github.com/pytorch/vision --quiet
pushd vision
pip install -q --user .
popd
rm -rf vision
}
install_torchvision
if [[ $PARALLEL == 1 ]]; then
pip install pytest-xdist
fi