文档首页 > > 用户指南> 自定义算子开发> 工程创建>

设置TBE Python库

设置TBE Python库

分享
更新时间:2020/08/03 GMT+08:00

算子开发之前,开发者需要参考本章节设置TBE算子工程依赖Python库。

设置全局依赖的Python SDK

  1. 在工程界面中,单击菜单栏中的File > Project Structure,进入Project Structure设置页面。
  2. 在左侧菜单栏中选择Platform Settings > SDKs,参考图1添加Python解释器。

    注意:从Interpreter后面的选择解析器版本,需要选择python3.7.5版本。
    图1 添加全局Python库
    System Interpreter表示使用本地Python解释器,若选用以下三种方式,您需要提前安装依赖并配置python镜像,安装依赖包括conda、pipenv等。
    1. Virtualenv Environment:使用虚拟管理环境,在创建Python虚拟环境的目录下创建独立的Python库和解释器,创建好后,会在创建目录下重新根据当前用户的~/.pip/pip.conf文件中配置的PIP源通过PIP工具进行库的下载。
    2. Conda Environment:选择Anaconda python解释器并利用conda创建虚拟环境。
    3. Pipenv Environment:整合了virtualenv与pip,可同时创建虚拟环境并进行库管理功能。

  3. 添加完成后,请确保算子开发相关Python库已存在,包括:te、topi等。

    如下图示例所示:

    图2 查看Python库

  4. Project Structure窗口中单击Apply。
  5. 单击OK,关闭Project Structure窗口。

设置当前算子工程依赖的Python SDK

  1. 在工程界面中,单击菜单栏中的File > Project Structure,进入Project Structure设置页面。
  2. 在左侧菜单栏中选择Project Settings > Project,进行Project设置。

    Project SDK中,下拉选择设置全局依赖的Python SDK中配置的Python SDK,并单击Apply,如图3所示。
    图3 选择Project SDK

  3. 在左侧菜单栏中选择Project Settings > Modules,进行Modules设置。

    Dependencies页签中,下拉选择设置TBE Python库中配置的Python SDK,并单击Apply,如图4所示。
    图4 选择Modules SDK

  4. 单击OK,关闭Project Structure窗口。

    此时,在工程底部菜单栏中会显示如图5所示提示。
    图5 更新界面

  5. 更新成功后,Python代码可以实现自动解析,如图6所示。单击导入的Python模块代码,然后按Ctrl+Q,会出现相应的Tips。按住Ctrl建,单击还可进入对应的模块定义文件。

    图6 Python代码自动解析示例

    若仍然无法自动解析,请关闭工程后重新打开。

(可选)添加自定义Python SDK

  1. 在工程界面中,单击菜单栏中的File > Project Structure,进入Project Structure设置页面。
  2. 在左侧菜单栏中选择Platform Settings > SDKs,进行Python SDK设置。
  3. 选择对应的python库,在Classpath栏中单击“+”,可添加自定义的python库,如图7所示。

    需要注意的是:

    • 在新建的Python库中,务必确保系统默认的算子开发相关Python库已存在,包括te、topi等。
    • 由于Platform Settings页签中的Python库为全局配置,对所有算子工程生效。
    图7 添加用户自定义Python库

  4. 单击OK,关闭Project Structure窗口。
分享:

    相关文档

    相关产品

文档是否有解决您的问题?

提交成功!非常感谢您的反馈,我们会继续努力做到更好!
反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区问答频道来与我们联系探讨

智能客服提问云社区提问