部署开发环境和运行环境,请参见《CANN 软件安装指南》对应Atlas产品的描述。
对于昇腾设备,已安装驱动、固件场景下,该环境可直接作为运行环境,执行编译生成的应用可执行文件。
需要根据运行环境的安装包,确定引用的组件目录,否则会导致运行报错。安装方案请参见《CANN 软件安装指南》。
安装CANN软件后,需要以CANN运行用户登录环境,执行source ${install_path}/set_env.sh命令设置环境变量,其中${install_path}为CANN软件的安装目录。
示例场景:例如板端环境上的可用Device数量为8,Device ID分别为:0、1、2、3、4、5、6、7。
# acl.rt.get_device_count接口获取到的可用Device数量为1,acl.rt.set_device(0)时,索引0对应的Device ID是1 export ASCEND_RT_VISIBLE_DEVICES = 1
# acl.rt.get_device_count接口获取到的可用Device数量为3,acl.rt.set_device(0)时,索引0对应的Device ID是2 export ASCEND_RT_VISIBLE_DEVICES = 2,3,4
# acl.rt.get_device_count接口获取到的可用Device数量为3,acl.rt.set_device(0)时,索引0对应的Device ID是4 export ASCEND_RT_VISIBLE_DEVICES = 4,3,2
# acl.rt.get_device_count接口获取到的可用Device数量为2,acl.rt.set_device(0)时,索引0对应的Device ID是2 export ASCEND_RT_VISIBLE_DEVICES = 2,3,-1,5
通过export命令,设置环境变量只在当前终端窗口生效,且只对设置环境变量之后启动的昇腾AI应用进程生效。
若将export命令写入“~/.bashrc”文件,使环境变量永久生效,则环境变量对该用户下的所有昇腾AI应用进程都生效。这种方式,可能会影响其它不需要调整Device ID的应用进程,请谨慎使用。将export命令写入“~/.bashrc”文件的方法如下: