Giters

JoePenna / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable Diffusion (https://arxiv.org/abs/2112.10752). Tweaks focused on training faces, objects, and styles.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Torch Install Failure: "raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled"

potesd opened this issue · comments

commented

After following the install guide to the letter, torch failed to compile with CUDA enabled.

Traceback (most recent call last):
  File "C:\Users\ShiftSpace\Dreambooth-Stable-Diffusion\main.py", line 138, in <module>
    dreambooth_config: JoePennaDreamboothConfigSchemaV1 = parse_arguments()
  File "C:\Users\ShiftSpace\Dreambooth-Stable-Diffusion\dreambooth_helpers\arguments.py", line 145, in parse_arguments
    config.saturate(
  File "C:\Users\ShiftSpace\Dreambooth-Stable-Diffusion\dreambooth_helpers\joepenna_dreambooth_config.py", line 143, in saturate
    self.validate_gpu_vram()
  File "C:\Users\ShiftSpace\Dreambooth-Stable-Diffusion\dreambooth_helpers\joepenna_dreambooth_config.py", line 159, in validate_gpu_vram
    gpu_vram = int(torch.cuda.get_device_properties(self.gpu).total_memory)
  File "C:\Users\ShiftSpace\.conda\envs\dreambooth_joepenna\lib\site-packages\torch\cuda\__init__.py", line 449, in get_device_properties
    _lazy_init()  # will define _get_device_properties
  File "C:\Users\ShiftSpace\.conda\envs\dreambooth_joepenna\lib\site-packages\torch\cuda\__init__.py", line 289, in _lazy_init
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled