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

This commit is contained in:
anzhengqi 2021-11-11 16:16:28 +08:00
parent b910870ecc
commit 18670623dd
1 changed files with 1 additions and 0 deletions

View File

@ -66,6 +66,7 @@ def test_resnet50_cifar10_gpu():
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")
os.system("nvidia-smi")
os.system("fuser -v /dev/nvidia*")
exec_network_shell = "cd resnet/scripts; sh run_distribute_train_gpu.sh {} {}" \
.format(dataset_path, config_path)
logger.warning("cmd [{}] is running...".format(exec_network_shell))