NPU编译器安装

1.安装环境

• Python 环境

  支持 Linux Python2.7、Python3.6、Python3.7

  注意: 若用户需要编译转换 PyTorch 模型，则 NPU 编译器版本至少 1.6.0b0 以上，且必须使用 Python3.7

• 前端深度学习框架的版本限制

  • TensorFlow 1.4 - 1.15
  • PyTorch 1.10 - 1.13

2.安装 gxnpuc 工具链

$ pip install npu_compiler

3.更新 gxnpuc 工具链

$ pip install --upgrade npu_compiler

4.查看工具链版本

安装或更新完成后，可以使用如下命令查看当前工具链的版本号。

$ gxnpuc --version

重要

如果运行 gxnpuc --version 出现如下的错误，请参考如下两个方法进行解决：

$ gxnpuc --version
bash: gxnpuc: command not found

• 方法一：确认 gxnpuc 文件是否在 ~/.local/bin 中，如果是，请将 ~/.local/bin 添加到 PATH 目录中，添加方法：在 ~/.bashrc 中添加如下语句

  export PATH=~/.local/bin:$PATH
  

• 方法二：通过 import npu_complier 查看 gxnpuc 的路径：

  $ python
  Python 3.6.5 |Anaconda, Inc.| (default, Apr 29 2018, 16:14:56)
  [GCC 7.2.0] on linux
  Type "help", "copyright", "credits" or "license" for more information.
  >>> import npu_compiler as nc
  >>> help(nc)
  

  输出如下log：

  Help on package npu_compiler:
  
  NAME
      npu_compiler - #coding: utf-8
  
  PACKAGE CONTENTS
      __main__
      v100 (package)
      v150 (package)
      v160 (package)
  
  DATA
      VERSION = '1.5.6'
  
  FILE
      /home/jindg/anaconda3/lib/python3.6/site-packages/npu_compiler/__init__.py
  

  最后将上面log中的 /home/jindg/anaconda3/lib 改为 /home/jindg/anaconda3/bin，然后将它添加到 ~/.bashrc 中的 PATH 中就可以了。

注意

如果还是解决不了问题，请联系国芯的工程师。