diff --git a/ci/scripts/github_skipped_tests_comment.py b/ci/scripts/github_skipped_tests_comment.py
index ef0630620b97..c07fbf4a8463 100755
--- a/ci/scripts/github_skipped_tests_comment.py
+++ b/ci/scripts/github_skipped_tests_comment.py
@@ -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
 
diff --git a/docker/install/ubuntu_install_caffe.sh b/docker/install/ubuntu_install_caffe.sh
index c37bfb764935..4d9763b69aa3 100755
--- a/docker/install/ubuntu_install_caffe.sh
+++ b/docker/install/ubuntu_install_caffe.sh
@@ -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
@@ -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