Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!26158 add command fuser to figure out the root cause of testcase failure 

Merge pull request !26158 from anzhengqi/optimize-testcase-resnet50
This commit is contained in:
i-robot 2021-11-15 09:04:09 +00:00 committed by Gitee
parent ba3ee7d8c1 18670623dd
commit f01b5e4816
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/st/model_zoo_tests/resnet50/test_resnet50_cifar10.py
View File

@ -66,6 +66,7 @@ def test_resnet50_cifar10_gpu():
dataset_path = os.path.join(utils.data_root, "cifar-10-batches-bin") dataset_path = os.path.join(utils.data_root, "cifar-10-batches-bin")
config_path = os.path.join(cur_model_path, "config", "resnet50_cifar10_config.yaml") config_path = os.path.join(cur_model_path, "config", "resnet50_cifar10_config.yaml")
os.system("nvidia-smi") os.system("nvidia-smi")
os.system("fuser -v /dev/nvidia*")
exec_network_shell = "cd resnet/scripts; sh run_distribute_train_gpu.sh {} {}" \ exec_network_shell = "cd resnet/scripts; sh run_distribute_train_gpu.sh {} {}" \
.format(dataset_path, config_path) .format(dataset_path, config_path)
logger.warning("cmd [{}] is running...".format(exec_network_shell)) logger.warning("cmd [{}] is running...".format(exec_network_shell))