Function verifyIsTorchInstalled

apps/node/src/modules/comfyui/verify-torch.ts:32–51

()

const tmpFile = path.resolve(getAppTmpDir(), "tmp.py");
export async function verifyIsTorchInstalled(): Promise<boolean> {
    const { PIP_PATH, PYTHON_PATH } = conda.getCondaPaths();
    try {
        await writeTmpPythonFile(pythonCode);
        // 使用 execa 执行 conda run 命令，切换到指定的 Conda 环境并运行 Python 代码
        const { stdout } = await runCommand(`${PYTHON_PATH} ${tmpFile}`);

        // 检查执行结果是否符合预期
        if (stdout.trim() === expectedOutput) {
            logger.info('Torch installation verification successful.');
            return true;
        } else {
            console.error('Torch installation verification failed. Unexpected output.');
            return false;
        }
    } catch (error) {
        console.error('Torch installation verification failed. Error:', error);
        return false;
    }
}

export async function writeTmpPythonFile(code: string) {
    await fsExtra.writeFile(tmpFile, code, 'utf8');