未验证 提交 1ee59750 编写于 作者: H huangjianhui 提交者: GitHub

Update util.py

delete pynvml and check_gpu_memory
上级 d8edeb3b
import os import os
import pynvml
import argparse import argparse
import base64 import base64
import subprocess import subprocess
...@@ -87,16 +86,6 @@ def kill_process(port, sleep_time=0): ...@@ -87,16 +86,6 @@ def kill_process(port, sleep_time=0):
# 解决端口占用 # 解决端口占用
os.system(f"sleep {sleep_time}") os.system(f"sleep {sleep_time}")
def check_gpu_memory(gpu_id):
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(gpu_id)
mem_info = pynvml.nvmlDeviceGetMemoryInfo(handle)
mem_used = mem_info.used / 1024 ** 2
print(f"GPU-{gpu_id} memory used:", mem_used)
return mem_used > 100
def count_process_num_on_port(port): def count_process_num_on_port(port):
command = "netstat -nlp | grep :" + str(port) + " | wc -l" command = "netstat -nlp | grep :" + str(port) + " | wc -l"
count = eval(os.popen(command).read()) count = eval(os.popen(command).read())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册