Skip to content

Error: Device type CUDA is not supported for torch.Generator() api #22

@Bunnnk

Description

@Bunnnk

"I am a beginner and encountered the following error. My current computer setup includes an AMD R7 6800H and an RTX 2050 GPU. The error message is as follows. I suspect the issue might be related to the hybrid graphics mode on my laptop, which doesn't use the dedicated GPU directly. Could this be the cause of the error? I would really appreciate your help."
:P

Pipelines loaded with `dtype=torch.float16` cannot run with `cpu` device. It is not recommended to move them to `cpu` as running them will fail. Please make sure to use an accelerator to run the pipeline in inference, due to the lack of support for`float16` operations on this device in PyTorch. Please, remove the `torch_dtype=torch.float16` argument, or use another device for inference.
Traceback (most recent call last):
  File "D:\StyleStudio\StyleStudio-main\infer_StyleStudio.py", line 123, in <module>
    main(args)
  File "D:\StyleStudio\StyleStudio-main\infer_StyleStudio.py", line 51, in main
    generator = torch.Generator("cuda").manual_seed(seed)
RuntimeError: Device type CUDA is not supported for torch.Generator() api.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions