levihsu
commented
2 months ago Please install the cuda version of pytorch
Closed wbsljy closed 2 months ago
levihsu
commented
2 months ago Please install the cuda version of pytorch
一开始运行.gradio_ootd.py文件时可以正常出现并访问本地web但无法生成图片，后面重新运行后直接报错： (ootd) D:\OOTDiffusion\run>python gradio_ootd.py Traceback (most recent call last): File "D:\OOTDiffusion\run\gradio_ootd.py", line 20, in
openpose_model_hd = OpenPose(0)
File "D:\OOTDiffusion\preprocess\openpose\run_openpose.py", line 32, in init
torch.cuda.set_device(gpu_id)
File "C:\anaconda3\envs\ootd\lib\site-packages\torch\cuda__init__.py", line 350, in set_device
torch._C._cuda_setDevice(device)
AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'