IT博客汇
  • 首页
  • 精华
  • 技术
  • 设计
  • 资讯
  • 扯淡
  • 权利声明
  • 登录 注册

    Intel extension for PyTorch

    RobinDong发表于 2023-05-26 00:43:49
    love 0

    Trying to test the Intel extension for PyTorch in my project, but it reported errors:

    Traceback (most recent call last):                                                                                                                                                                                                          
      File "reviewjpgs_optimaztion_testing.py", line 27, in <module>                                                                                                                                                                            
        import intel_extension_for_pytorch as ipex                                                                                                                                                                                              
      File "/home/hero/.pyenv/versions/3.8.12/lib/python3.8/site-packages/intel_extension_for_pytorch/__init__.py", line 11, in <module>                                                                                                        
        from .cpu import _cpu_isa                                                                                                                                                                                                               
      File "/home/hero/.pyenv/versions/3.8.12/lib/python3.8/site-packages/intel_extension_for_pytorch/cpu/__init__.py", line 1, in <module>                                                                                                     
        from . import runtime                                                                                                                                                                                                                   
      File "/home/hero/.pyenv/versions/3.8.12/lib/python3.8/site-packages/intel_extension_for_pytorch/cpu/runtime/__init__.py", line 3, in <module>                                                                                             
        from .multi_stream import MultiStreamModule, get_default_num_streams, \
      File "/home/hero/.pyenv/versions/3.8.12/lib/python3.8/site-packages/intel_extension_for_pytorch/cpu/runtime/multi_stream.py", line 4, in <module>
        import intel_extension_for_pytorch._C as core
    ImportError: /home/hero/.pyenv/versions/3.8.12/lib/python3.8/site-packages/intel_extension_for_pytorch/lib/libintel-ext-pt-cpu.so: undefined symbol: _ZNK3c1010TensorImpl22is_strides_like_customENS_12MemoryFormatE

    The answer is quite tricky: need to install the IPEX package with the same version of PyTorch.

    After the testing of both torch.jit.trace and this IPEX, we found out that `torch.jit.trace` could boost the performance of prediction significantly but IPEX could not.



沪ICP备19023445号-2号
友情链接