!16235 make device_id can be getted from environment and modify required version for package pycocotools

From: @zhouneng2
Reviewed-by: @oacjiewen,@liangchenghui
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2021-05-12 09:34:44 +08:00 committed by Gitee
commit fdd0c73cd5
2 changed files with 4 additions and 5 deletions

View File

@ -64,9 +64,8 @@ if __name__ == '__main__':
else:
rank = 0
group_size = 1
context.set_context(device_id=0)
# if os.getenv('DEVICE_ID', "not_set").isdigit():
# context.set_context(device_id=int(os.getenv('DEVICE_ID')))
device_id = int(os.getenv('DEVICE_ID', '0'))
context.set_context(device_id=device_id)
context.set_context(mode=context.GRAPH_MODE, device_target=args_opt.device_target, save_graphs=False)
# define network

View File

@ -15,7 +15,7 @@ sklearn >= 0.0 # for st test
pandas >= 1.0.2 # for ut test
astunparse >= 1.6.3
packaging >= 20.0
pycocotools >= 2.0.0 # for st test
pycocotools >= 2.0.2 # for st test
tables >= 3.6.1 # for st test
psutil >= 5.6.1
subword-nmt>=0.3.7 # for st test