def METHOD_NAME(self, cuda_version):
    if cuda_version:
        if platform.system() == "Darwin":
            print(
                "CUDA not supported on MacOS. Refer https://pytorch.org/ for installing from source."
            )
            sys.exit(1)
        elif cuda_version == "cu92" and platform.system() == "Windows":
            print(
                "CUDA 9.2 not supported on Windows. Refer https://pytorch.org/ for installing from source."
            )
            sys.exit(1)
        else:
            os.system(
                f"{sys.executable} -m pip install -U -r requirements/torch_{cuda_version}_{platform.system().lower()}.txt"
            )
    elif args.neuronx:
        torch_neuronx_requirements_file = os.path.join(
            "requirements", "torch_neuronx_linux.txt"
        )
        os.system(
            f"{sys.executable} -m pip install -U -r {torch_neuronx_requirements_file}"
        )
    else:
        os.system(
            f"{sys.executable} -m pip install -U -r requirements/torch_{platform.system().lower()}.txt"
        )