Update CUDA Graph status output

This commit is contained in:
RVC-Boss
2026-07-20 23:13:44 +08:00
parent b3f54df1be
commit 8f2fdbf483

View File

@@ -143,7 +143,9 @@ print(
i18n("当前设备:%s | 推理精度:%s") % (config.device, config.dtype), i18n("当前设备:%s | 推理精度:%s") % (config.device, config.dtype),
flush=True, flush=True,
) )
logger.info("RVC_CUDA_GRAPH=%s", os.environ.get("RVC_CUDA_GRAPH", "0")) cuda_graph_status = os.environ.get("RVC_CUDA_GRAPH", "0")
print(f"RVC_CUDA_GRAPH={cuda_graph_status}", flush=True)
logger.info("RVC_CUDA_GRAPH=%s", cuda_graph_status)
# GPU filtering and precision rules are shared with inference/extraction/training. # GPU filtering and precision rules are shared with inference/extraction/training.
gpu_infos = list(GPU_INFOS) gpu_infos = list(GPU_INFOS)
gpu_indices = sorted(GPU_INDEX) gpu_indices = sorted(GPU_INDEX)