Function installPyTorchForGPU

apps/node/src/modules/comfyui/bootstrap.ts:296–373

(dispatcher: TaskEventDispatcher, nightly: boolean = false)

 * @returns 
 */
export async function installPyTorchForGPU(dispatcher: TaskEventDispatcher, nightly: boolean = false): Promise<boolean> {
    const { PIP_PATH, PYTHON_PATH } = conda.getCondaPaths();
    let installCommand = "";
    logger.info("start installing Pytorch");
    dispatcher({
        message: "Start installing PyTorch..."
    });

    let success = false;
    let lastError = null;
    for (let i = 0; i < 3; i++) {
        if (isMac) {
            try {
                installCommand = `${PIP_PATH} install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu`;
                await runCommand(installCommand, dispatcher);
                success = true;
                break;
            } catch (error: any) {
                lastError = error;
            }
        } else {
            try {
                const gpuType = await getGPUType()

                dispatcher({
                    message: "GPU Type: " + gpuType + ", pip: " +  PIP_PATH
                })
                // AMD GPU
                if (gpuType === 'amd') {
                    const rocmVersion = nightly ? 'rocm5.7' : 'rocm5.6';
                    const installCommand = nightly
                        ? `${PIP_PATH} install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/${rocmVersion}`
                        : ` ${PIP_PATH} install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/${rocmVersion}`;

                    await runCommand(installCommand, dispatcher);
                    success = true;
                    break;
                }
                // NVIDIA GPU
                else if (gpuType === 'nvidia') {
                    installCommand = `${PIP_PATH} install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121`;
                    await runCommand(installCommand, dispatcher);
                    success = true;
                    break;
                } 

                // https://github.com/Comfy-Org/comfy-cli/issues/50
                else if (gpuType === "intel") {
                    await runCommand(`${PIP_PATH} install mkl mkl-dpcpp`, dispatcher);
                    installCommand = `${PIP_PATH} install torch==2.1.0.post2 torchvision==0.16.0.post2 torchaudio==2.1.0.post2 intel-extension-for-pytorch==2.1.30 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/`
                    await runCommand(installCommand, dispatcher);
                    success = true;
                    break;
                }

                else {
                    installCommand = `${PIP_PATH} install torch torchvision torchaudio`;
                    await runCommand(installCommand, dispatcher);