diff --git a/colossalai/cli/benchmark/__init__.py b/colossalai/cli/benchmark/__init__.py
index 3653f1ffa..c020d33b6 100644
--- a/colossalai/cli/benchmark/__init__.py
+++ b/colossalai/cli/benchmark/__init__.py
@@ -1,4 +1,3 @@
-from random import choices
 import click
 
 from .utils import *
diff --git a/colossalai/cli/benchmark/utils.py b/colossalai/cli/benchmark/utils.py
index 0317f2a41..41ac38b82 100644
--- a/colossalai/cli/benchmark/utils.py
+++ b/colossalai/cli/benchmark/utils.py
@@ -1,10 +1,8 @@
 import math
 import time
-from grpc import Call
 import torch
 
 from colossalai.utils import MultiTimer
-from colossalai.core import global_context as gpc
 from colossalai.context import ParallelMode, Config
 from typing import List, Dict, Tuple, Callable
 
diff --git a/colossalai/cli/launcher/multinode_runner.py b/colossalai/cli/launcher/multinode_runner.py
index b3cabcbf2..c45ad5e5a 100644
--- a/colossalai/cli/launcher/multinode_runner.py
+++ b/colossalai/cli/launcher/multinode_runner.py
@@ -1,5 +1,4 @@
 import fabric
-from fabric import Connection
 from .hostinfo import HostInfo, HostInfoList
 from multiprocessing import Pipe, Process
 from multiprocessing import connection as mp_connection