CANN软件提供进程级环境变量设置脚本,供用户在进程中引用,以自动完成环境变量设置。用户进程结束后自动失效。
可在程序启动的Shell脚本中使用如下命令设置CANN的相关环境变量,也可通过命令行执行如下命令(以root用户默认安装路径“/usr/local/Ascend”为例):
source /usr/local/Ascend/ascend-toolkit/set_env.sh source /usr/local/Ascend/tfplugin/set_env.sh
# 解压到当前目录下 tar -xf Ascend-mindxsdk-mxrec-{version}_linux-{arch}.tar.gz --no-same-owner # 解压到特定目录下 tar -xf Ascend-mindxsdk-mxrec-{version}_linux-{arch}.tar.gz --no-same-owner -C 指定路径
mindxsdk-mxrec/ |-- cust_op #自定义算子 | |-- cust_op_by_addr | `-- fused_lazy_adam |-- examples #样例模型 | |-- DCNv2 | |-- WideDeep | |-- demo | |-- dlrm | |-- ps_adapt_to_mxrec | |-- rec_infer | `-- xDeepFM |-- tf1_whl | `-- mx_rec-{version}-py3-none-{arch}.whl #适用于TensorFlow 1.15.0版本的mx_rec框架Wheel包 |-- tf2_whl | `-- mx_rec-{version}-py3-none-{arch}.whl #适用于TensorFlow 2.6.5版本的mx_rec框架Wheel包 `-- version.info
pip3 install mx_rec-{version}-py3-none-{arch}.whl
export PYTHONPATH={mxrec_install_path}:{mxrec_install_path}/mxRec:$PYTHONPATH
pip3.7 install numpy decorator sympy==1.4 cffi==1.12.3 pyyaml pathlib2 grpcio grpcio-tools protobuf==3.20.0 scipy requests mpi4py easydict scikit-learn==0.20.0 attrs
HOROVOD_WITH_MPI=1 HOROVOD_WITH_TENSORFLOW=1 pip3.7 install horovod --no-cache-dir
bash run.sh