×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

是否支持stable-diffusion yue_water2024-01-03 11:29:02 回复 1 查看 使用求助 经验交流
是否支持stable-diffusion
分享到:

安装Cambricon PyTorch后,通过安装stable-diffusion,提示Torch is not able to use GPU;请问有什么好的解决办法?

GPU型号:MLU370-M8



【使用操作系统】必填*:
例如:ubuntu 20.04


【使用驱动版本】必填*:

+------------------------------------------------------------------------------+

| CNMON v5.10.12                                               Driver v5.10.12 |

+-------------------------------+----------------------+-----------------------+

| Card  VF  Name       Firmware |               Bus-Id | Util        Ecc-Error |

| Fan   Temp      Pwr:Usage/Cap |         Memory-Usage | SR-IOV   Compute-Mode |

|===============================+======================+=======================|

| 0     /   MLU370-M8    v1.1.4 |         0000:A9:00.0 | 0%                N/A |

|  0%   26C         53 W/ 300 W | 28492 MiB/ 47868 MiB | N/A           Default |

+-------------------------------+----------------------+-----------------------+

| 1     /   MLU370-M8    v1.1.4 |         0000:D2:00.0 | 0%                N/A |

|  0%   24C         50 W/ 300 W |     0 MiB/ 47868 MiB | N/A           Default |

+-------------------------------+----------------------+-----------------------+


+------------------------------------------------------------------------------+

| Processes:                                                                   |

|  Card  MI  PID     Command Line                             MLU Memory Usage |

|==============================================================================|

|  0     /   18976   /torch/venv3/pytorch/bin/python                 28396 MiB |

+------------------------------------------------------------------------------+

【当前已做了哪些信息确认】:
例如:已经装过Cambricon PyTorch 1.13


【出错信息】*:Torch is not able to use GPU

Traceback (most recent call last):

  File "/stable_diffusion/Project/stable-diffusion-webui/launch.py", line 48, in <module>

    main()

  File "/stable_diffusion/Project/stable-diffusion-webui/launch.py", line 39, in main

    prepare_environment()

  File "/stable_diffusion/Project/stable-diffusion-webui/modules/launch_utils.py", line 384, in prepare_environment

    raise RuntimeError(

RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check





 


版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭