Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions ci/scripts/github_skipped_tests_comment.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,6 @@
from urllib import error
from xml.etree import ElementTree

import requests

from git_utils import git, GitHubRepo, parse_remote
from cmd_utils import init_log

Expand Down
8 changes: 7 additions & 1 deletion docker/install/ubuntu_install_caffe.sh
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,11 @@

set -euxo pipefail

if [ -z "${TVM_VENV+x}" ]; then
echo "ERROR: expect TVM_VENV env var to be set"
exit 2
fi

apt-get update --fix-missing

# # Install dependencies
Expand Down Expand Up @@ -60,4 +65,5 @@ cd / && rm -rf /caffe_src

PYCAFFE_ROOT=${CAFFE_HOME}/python
echo "${CAFFE_HOME}/lib" >> /etc/ld.so.conf.d/caffe.conf && ldconfig
ln -s ${PYCAFFE_ROOT}/caffe /usr/local/lib/python3.7/dist-packages/caffe
VENV_SITE_PACKAGE=$(pip3 show numpy | grep "Location:" | cut -d ' ' -f 2)
ln -s ${PYCAFFE_ROOT}/caffe ${VENV_SITE_PACKAGE}/caffe