专栏算法工具链J6镜像提示no module named horizon_plugin_pytorch

J6镜像提示no module named horizon_plugin_pytorch

已解决
chengang2025-03-27
60
4
你们好， 最近在使用v3.0.31版J6镜像时，训练模型时遇到一个报错，具体是执行import horizon_pytorch as horizon，报错显示 no module named horizon_plugin_pytorch？这是镜像问题吗，但是v.3.0.17就没有问题。 
算法工具链
征程6
评论2
0/1000
  • 费小财
    Lv.5

    应该是去掉了一个库，装一个库就解决了

    2025-03-27
    0
    0
  • DR_KAN
    Lv.4

    要用gpu docker，cpu docker没有QAT环境，可以检查一下是不是GPU的

    2025-03-28
    0
    2
    • chengang回复DR_KAN:

      GPU版docker镜像要商务获取是吗？

      2025-03-28
      0
    • DR_KAN回复chengang:

      你找谁获取的cpu docker，就找谁获取gpu docker

      2025-03-28
      0