运行报错，在哪里修改占用显卡的个数呢？

IDEA-CCNL / Taiyi-Diffusion-XL

Taiyi-Diffusion-XL训练代码

Apache License 2.0

21 stars 4 forks source link

运行报错，在哪里修改占用显卡的个数呢？ #3

Closed yangy996 closed 7 months ago

yangy996 commented 7 months ago

RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

wxj630 commented 7 months ago

config/accelerate_config/default_config.yaml

yangy996 commented 7 months ago

config/accelerate_config/default_config.yaml

多谢多谢