03b9c4930a
* env script * update README.md * modify README * modify cpu info output * add env-check.sh * add env-check.bat * add windows * modify bat
64 lines
No EOL
1.7 KiB
Python
64 lines
No EOL
1.7 KiB
Python
import sys
|
|
|
|
|
|
def check_python_version():
|
|
py_version = sys.version.split()[0]
|
|
lst = py_version.split(".")
|
|
v1 = eval(lst[0])
|
|
v2 = eval(lst[1])
|
|
if v1!=3 or v2<9:
|
|
print("Python Version must be higher than 3.9.x, please check python version. More details could be found in the README.md")
|
|
return 1
|
|
return 0
|
|
|
|
def check_transformer_version():
|
|
try:
|
|
import transformers
|
|
print(f"transformers={transformers.__version__}")
|
|
except:
|
|
print("Transformers is not installed.")
|
|
|
|
def check_torch_version():
|
|
try:
|
|
import torch
|
|
print(f"torch={torch.__version__}")
|
|
except:
|
|
print("PyTorch is not installed.")
|
|
|
|
|
|
def check_torch_version():
|
|
try:
|
|
import torch
|
|
print(f"torch={torch.__version__}")
|
|
except:
|
|
print("PyTorch is not installed.")
|
|
|
|
def check_bigdl_version():
|
|
import os
|
|
if os.system("pip show bigdl-llm")!=0:
|
|
print("BigDL is not installed")
|
|
|
|
|
|
def check_ipex_version():
|
|
try:
|
|
import intel_extension_for_pytorch as ipex
|
|
print(f"ipex={ipex.__version__}")
|
|
except:
|
|
print("IPEX is not installed properly. ")
|
|
|
|
def main():
|
|
if check_python_version()!=0:
|
|
return -1
|
|
print("-----------------------------------------------------------------")
|
|
check_transformer_version()
|
|
print("-----------------------------------------------------------------")
|
|
check_torch_version()
|
|
print("-----------------------------------------------------------------")
|
|
check_bigdl_version()
|
|
print("-----------------------------------------------------------------")
|
|
check_ipex_version()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |