¿Cómo uso varias tarjetas de Ascend para la depuración en una instancia de notebook?

Un trabajo de entrenamiento multi-tarjeta de Ascend se ejecuta en modo de multi-proceso y multi-tarjeta. El número de tarjetas es igual al número de procesos de Python. El subcapa de Ascend lee la variable de entorno RANK_TABLE_FILE, que se ha configurado en el entorno de desarrollo, sin necesidad de configuración manual. Por ejemplo, para ejecutar un trabajo en ocho tarjetas, el código es el siguiente:

 export RANK_SIZE=8
 current_exec_path=$(pwd)
 echo 'start training'
 for((i=0;i<=$RANK_SIZE-1;i++));
 do
 echo 'start rank '$i
 mkdir ${current_exec_path}/device$i
 cd ${current_exec_path}/device$i
 echo $i
 export RANK_ID=$i
 dev=`expr $i + 0`
 echo $dev
 export DEVICE_ID=$dev
 python train.py > train.log 2>&1 &
 done

Set the environment variable DEVICE_ID in train.py.

devid = int(os.getenv('DEVICE_ID'))
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", device_id=devid)