frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move torchvision install out of onnx test script

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/20890

Differential Revision: D15486657

Pulled By: bddppq

fbshipit-source-id: 3acd7386d1f070cad9bd43d6e74244b706c0dc16
This commit is contained in:
Junjie Bai
2019-05-23 17:59:33 -07:00
committed by Facebook Github Bot
parent d5b7138a2c
commit bd53c8eb93
2 changed files with 1 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.jenkins/caffe2/test.sh
View File

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

14
scripts/onnx/test.sh
View File

@ -24,20 +24,6 @@ done
set -- "${UNKNOWN[@]}" # leave UNKNOWN set -- "${UNKNOWN[@]}" # leave UNKNOWN
pip install pytest scipy 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 if [[ $PARALLEL == 1 ]]; then
pip install pytest-xdist pip install pytest-xdist
fi fi