Skip to content

Commit ea45b6b

Browse files
authored
Merge pull request #2 from colesbury/vision
Install torchvision before building docs
2 parents 34fcbd8 + 1f1e285 commit ea45b6b

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

jenkins/pytorch/build_nimbix.sh

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,15 @@ time python setup.py install
158158
echo "Testing pytorch"
159159
time test/run_test.sh
160160

161+
cd ..
162+
echo "Installing torchvision at branch master"
163+
rm -rf vision
164+
git clone https://pytorchbot:$GITHUB_TOKEN@github.com/pytorch/vision --quiet
165+
cd vision
166+
pip install -r requirements.txt || true
167+
time python setup.py install
168+
cd ..
169+
161170
echo "ALL CHECKS PASSED"
162171

163172
if [ "$GIT_BRANCH" == "origin/master" ]

0 commit comments

Comments
 (0)