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 指定路径
1 2 3 4 5 6 | mindxsdk-mxrec/ |-- tf1_whl | `-- mx_rec-{version}-py3-none-linux_{arch}.whl #适用于TensorFlow 1.15.0版本的mx_rec框架Wheel包 |-- tf2_whl | `-- mx_rec-{version}-py3-none-linux_{arch}.whl #适用于TensorFlow 2.6.5版本的mx_rec框架Wheel包 `-- version.info |
pip3 install mx_rec-{version}-py3-none-linux_{arch}.whl
export PYTHONPATH={rec_install_path}:{rec_install_path}/mx_rec:$PYTHONPATH
1 | 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