levihsu / OOTDiffusion

Official implementation of OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on
Other
5.07k stars 740 forks source link

multi gpu error #169

Closed TigerHH6866 closed 2 months ago

TigerHH6866 commented 2 months ago

i have 2 gpus 3080Ti

log

(ootd) root@autodl-container-71164ea666-32411d06:~/autodl-tmp/OOTDiffusion-main/run# python gradio_ootd.py Loading pipeline components...: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 7/7 [00:01<00:00, 6.67it/s] text_config_dict is provided which will be used to initialize CLIPTextConfig. The value text_config["id2label"] will be overriden. text_config_dict is provided which will be used to initialize CLIPTextConfig. The value text_config["bos_token_id"] will be overriden. text_config_dict is provided which will be used to initialize CLIPTextConfig. The value text_config["eos_token_id"] will be overriden. Traceback (most recent call last): File "/root/autodl-tmp/OOTDiffusion-main/run/gradio_ootd.py", line 25, in openpose_model_dc = OpenPose(1) File "/root/autodl-tmp/OOTDiffusion-main/preprocess/openpose/run_openpose.py", line 32, in init torch.cuda.set_device(gpu_id) File "/root/miniconda3/envs/ootd/lib/python3.10/site-packages/torch/cuda/init.py", line 351, in set_device torch._C._cuda_setDevice(device) 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.