环境准备
- 使用ModelSlim工具前,需参考《CANN 软件安装指南》搭建开发环境。
- 安装CANN软件后,需要以CANN运行用户登录环境,执行如下示例命令配置环境变量。
source {CANN包安装路径}/ascend-toolkit/set_env.sh
- 使用非root用户运行调优任务时,需要管理员用户将运行用户加入驱动运行用户组(例如:HwHiAiUser)中,保证普通用户对run包的lib库有读权限。
- ModelSlim工具依赖Python,以Python3.7.5为例,请以运行用户执行如下命令设置Python3.7.5的相关环境变量。
#用于设置python3.7.5库文件路径 export LD_LIBRARY_PATH=/usr/local/python3.7.5/lib:$LD_LIBRARY_PATH #如果用户环境存在多个python3版本,则指定使用python3.7.5版本 export PATH=/usr/local/python3.7.5/bin:$PATH
Python3.7.5安装路径请根据实际情况进行替换,您也可以将以上命令写入~/.bashrc文件中,然后执行source ~/.bashrc命令使其立即生效。
- 用户根据需要自行安装以下AI框架包(请注意CANN与MindSpore、PyTorch的版本配套关系):
父主题: ModelSlim工具