diff --git a/torch_npu/npu/_memory_viz.py b/torch_npu/npu/_memory_viz.py
index 3d02d071ce876688cbadad2152b610f7ec4522de..bdb6dd1d6a50fbdea8c36e02065f427cdcf9786c 100644
--- a/torch_npu/npu/_memory_viz.py
+++ b/torch_npu/npu/_memory_viz.py
@@ -1,7 +1,6 @@
 import sys
 import os
 import io
-import subprocess
 import json
 from functools import lru_cache
 from typing import Any