Commit 2110591b authored by dmMaze's avatar dmMaze
Browse files

fix #524

parent 82d2df27

launch.py

+1 −2
Original line number Diff line number Diff line
@@ -288,8 +288,7 @@ def prepare_environment(): 
            except Exception:

                run_pip(f"install {req}", req)

                req_updated = True



    torch_command = os.environ.get('TORCH_COMMAND', "pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 --disable-pip-version-check")

    torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu118 --disable-pip-version-check")

    if args.reinstall_torch or not is_installed("torch") or not is_installed("torchvision"):

        run(f'"{python}" -m {torch_command}', "Installing torch and torchvision", "Couldn't install torch", live=True)

        req_updated = True