diff --git a/ci/docker/Dockerfile.test.arm b/ci/docker/Dockerfile.test.arm
index c02f4b73e422..bd118f63f097 100644
--- a/ci/docker/Dockerfile.test.arm
+++ b/ci/docker/Dockerfile.test.arm
@@ -35,11 +35,11 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \
 # Python dependencies
 RUN pip3 install --no-cache-dir --upgrade pip && \
     pip3 install --no-cache-dir \
-    pytest==5.3.5 \
+    pytest==6.1.2 \
     pytest-env==0.6.2 \
-    pytest-cov==2.8.1 \
-    pytest-xdist==1.31.0 \
-    pytest-timeout==1.3.4 \
+    pytest-cov==2.10.1 \
+    pytest-xdist==2.1.0 \
+    pytest-timeout==1.4.2 \
     mock==2.0.0
 
 ARG USER_ID=0
diff --git a/ci/docker/install/requirements b/ci/docker/install/requirements
index e7e6fd4e8ccf..0654fe287668 100644
--- a/ci/docker/install/requirements
+++ b/ci/docker/install/requirements
@@ -36,15 +36,14 @@ Cython==0.29.7
 cpplint==1.3.0
 pylint==2.3.1  # pylint and astroid need to be aligned
 astroid==2.3.3  # pylint and astroid need to be aligned
-pytest==5.3.5
+pytest==6.1.2
 pytest-env==0.6.2
-pytest-cov==2.8.1
-pytest-xdist==1.31.0
-pytest-timeout==1.3.4
-flaky==3.6.1
+pytest-cov==2.10.1
+pytest-xdist==2.1.0
+pytest-timeout==1.4.2
+flaky==3.7.0
 setuptools==49.6.0  # https://github.com/pypa/setuptools/issues/2352
 wheel
-mock==2.0.0
 
 # TVM dependencies
 decorator==4.4.0
diff --git a/docker/install/python.sh b/docker/install/python.sh
index d49b500d843c..44011f822a2a 100755
--- a/docker/install/python.sh
+++ b/docker/install/python.sh
@@ -24,4 +24,4 @@ apt-get update && apt-get install -y python3-dev
 # the version of the pip shipped with ubuntu may be too lower, install a recent version here
 cd /tmp && wget https://bootstrap.pypa.io/get-pip.py && python3 get-pip.py
 
-pip3 install pylint numpy requests Pillow pytest==5.3.2 pytest-env==0.6.2 pytest-cov==2.8.1 pytest-xdist==1.31.0
+pip3 install pylint numpy requests Pillow pytest==6.1.2 pytest-env==0.6.2 pytest-cov==2.10.1 pytest-xdist==2.1.0