下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

配置环境变量

  1. 准备数据集并上传到运行环境的目录下,例如:/home/data/resnet50/imagenet。
  2. 请配置如下训练相关环境变量shell脚本env.sh,再拉起训练。
    # 配置CANN相关环境变量
    CANN_INSTALL_PATH_CONF='/etc/Ascend/ascend_cann_install.info'
    
    if [ -f $CANN_INSTALL_PATH_CONF ]; then
      DEFAULT_CANN_INSTALL_PATH=$(cat $CANN_INSTALL_PATH_CONF | grep Install_Path | cut -d "=" -f 2)
    else
      DEFAULT_CANN_INSTALL_PATH="/usr/local/Ascend/"
    fi
    
    CANN_INSTALL_PATH=${1:-${DEFAULT_CANN_INSTALL_PATH}}
    
    if [ -d ${CANN_INSTALL_PATH}/ascend-toolkit/latest ];then
      source ${CANN_INSTALL_PATH}/ascend-toolkit/set_env.sh
    else
      source ${CANN_INSTALL_PATH}/nnae/set_env.sh
    fi
    
    # 导入依赖库
    export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/openblas/lib
    export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib/
    export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/lib64/
    export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/lib/
    export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/lib/aarch64_64-linux-gnu
    
    # 配置自定义环境变量
    export HCCL_WHITELIST_DISABLE=1
    
    # log
    export ASCEND_SLOG_PRINT_TO_STDOUT=0   # 日志打屏, 可选
    export ASCEND_GLOBAL_LOG_LEVEL=3       # 日志级别常用 1 INFO级别; 3 ERROR级别
    export ASCEND_GLOBAL_EVENT_ENABLE=0    # 默认不使能event日志信息
  3. 执行脚本,配置环境变量。
    source env.sh

不支持float16类型的inf/nan数据输入输出。inf表示无穷大数值,nan表示无穷小数值,当一个数超出表示范围时会转为inf,小于能表示的精度时会转为nan。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词