下载
中文
注册

配置环境变量

PyTorch在线推理所依赖的环境变量和配置如下:

  • 请依据实际在下列场景中选择其一,进行在线推理依赖包安装路径的环境变量设置。具体如下(以HwHiAiUser用户安装,安装路径为默认路径为例):
    • 场景一:昇腾设备安装部署开发套件包Ascend-cann-toolkit(此时开发环境可进行推理任务)。
      . /home/User/Ascend/ascend-toolkit/set_env.sh
    • 场景二:昇腾设备安装部署软件包Ascend-cann-nnae。此时需要参考CANN 软件安装指南中的“安装运行环境(nnae软件,在物理机安装)”章节安装Ascend-cann-nnae。
      . /home/HwHiAiUser/Ascend/nnae/set_env.sh
  • 若运行环境中存在多个Python3版本时,需要在环境变量中配置指定Python版本的安装路径。以Python3.7.5为例:
    export PATH=/usr/local/python3.7.5/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/python3.7.5/lib:$LD_LIBRARY_PATH
  • 指定芯片的逻辑ID。
    export ASCEND_RT_VISIBLE_DEVICES=0
  • 输出日志信息,可根据实际修改。
    export ASCEND_SLOG_PRINT_TO_STDOUT=0
    export ASCEND_GLOBAL_LOG_LEVEL=3
表1 环境变量说明表

配置项

说明

必选/可选

LD_LIBRARY_PATH

动态库的查找路径,参考上述举例配置。

说明:

若系统环境安装了gcc7.3.0(例如CentOS7.6、Debian和BClinux系统),需要配置gcc相关环境变量。

export LD_LIBRARY_PATH=${install_path}/lib64:${LD_LIBRARY_PATH} 

其中${install_path}为gcc7.3.0安装路径。

必选

PATH

可执行程序的查找路径,参考上述举例配置。

必选

ASCEND_RT_VISIBLE_DEVICES

指定芯片的逻辑ID。取值范围[0,N-1],默认为0。其中N为当前物理机/虚拟机/容器内的设备总数。

可选

ASCEND_SLOG_PRINT_TO_STDOUT

是否开启日志打屏。

  • 0或不配置:关闭日志打屏
  • 1:开启日志打屏

可选

ASCEND_GLOBAL_LOG_LEVEL

设置日志的全局日志级别。

  • 0:对应DEBUG级别。
  • 1:对应INFO级别。
  • 2:对应WARNING级别。
  • 3:对应ERROR级别。
  • 4:对应NULL级别,不输出日志。
  • 其他值为非法值。

可选

更多日志信息,请参见日志参考