|
|
|
@@ -34,6 +34,7 @@ def parse_args(): |
|
|
|
args = parser.parse_args() |
|
|
|
return args |
|
|
|
|
|
|
|
|
|
|
|
def check_deps_version(mindspore_version, supported_version): |
|
|
|
""" |
|
|
|
check te/hccl/topi version |
|
|
|
@@ -62,17 +63,18 @@ def check_deps_version(mindspore_version, supported_version): |
|
|
|
print(f"MindSpore version {mindspore_version} and \"topi\" wheel package version {v} does not " |
|
|
|
"match, reference to the match info on: https://www.mindspore.cn/install") |
|
|
|
|
|
|
|
# pylint: disable=broad-except |
|
|
|
except Exception as e: |
|
|
|
print("CheckFailed: ", e.args) |
|
|
|
print("Minspore relies on the 3 whl packages of \"te\", \"topi\" and \"hccl\" in the \"fwkacllib\" " |
|
|
|
"folder of the Ascend 910 AI software package, please check whether they are installed " |
|
|
|
"correctly or not, reference to the match info on: https://www.mindspore.cn/install") |
|
|
|
|
|
|
|
|
|
|
|
def main(): |
|
|
|
args = parse_args() |
|
|
|
check_deps_version(args.mindspore_version, args.supported_version) |
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
sys.path = sys.path[1:] # avoid the impact of relative path env, only affect this process |
|
|
|
main() |