Browse Source

!187 Gpu environment check

From: @wilfchen
Reviewed-by: @zhoufeng54,@xu-yfei,@zhoufeng54,@xu-yfei
Signed-off-by: @xu-yfei
tags/v1.2.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
da10fbbd32
2 changed files with 4 additions and 3 deletions
  1. +3
    -2
      mindspore_serving/worker/_check_version.py
  2. +1
    -1
      third_party/mindspore

+ 3
- 2
mindspore_serving/worker/_check_version.py View File

@@ -250,12 +250,13 @@ class GPUEnvChecker():
"""Get gpu lib path by ldd command."""
path_list = []
current_path = os.path.split(os.path.realpath(__file__))[0]
ldd_result = subprocess.run(["ldd " + current_path + "/_c_expression*.so* | grep " + lib_name],
mindspore_path = os.path.dirname(os.path.dirname(current_path)) + "/mindspore"
ldd_result = subprocess.run(["ldd " + mindspore_path + "/_c_expression*.so* | grep " + lib_name],
timeout=3, text=True, capture_output=True, check=False, shell=True)
if ldd_result.returncode:
logger.warning(f"{lib_name} so(need by mndspore-gpu) is not found, please confirm that "
f"_c_experssion.so depend on {lib_name}, "
f"and _c_expression.so in directory:{current_path}")
f"and _c_expression.so in directory:{mindspore_path}")
return path_list
result = ldd_result.stdout
for i in result.split('\n'):


+ 1
- 1
third_party/mindspore

@@ -1 +1 @@
Subproject commit 363e574ff8c7b6b7a695c05e0c34086c177e5d41
Subproject commit bc6ad2127863f9f7fd7b2f9b25841e508b99fd51

Loading…
Cancel
Save