目录
git地址:
运行必须参数设置
git地址:
https://github.com/huggingface/optimum-habana
git clone 以后
cd optimum-habana
pip install .
运行必须参数设置
hpu改为 cuda,
output = subprocess.run(
"pip list | grep habana-torch-plugin",
shell=True,
text=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
)
return version.parse(output.stdout.split("\n")[0].split()[-1])