df9aba0c6e
- 将 scan_doc_dir.py 重命名为 document.py - 将 socket.py 重命名为 port.py - 更新 __init__.py 中的导入语句 - 在 port.py 中添加测试代码，用于查找可用端口
15 lines
561 B
Python
15 lines
561 B
Python
import socket
|
|
|
|
# 启动 TensorBoard 子进程前添加端口检测逻辑
|
|
def find_available_port(start_port):
|
|
port = start_port
|
|
while True:
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|
if s.connect_ex(('localhost', port)) != 0: # 端口未被占用
|
|
return port
|
|
port += 1 # 如果端口被占用，尝试下一个端口
|
|
|
|
if __name__ == "__main__":
|
|
start_port = 6006 # 起始端口号
|
|
available_port = find_available_port(start_port)
|
|
print(f"Available port: {available_port}") |