Open eliranwong opened 2 weeks ago
I can see onnx runtime support AMD rocm, please read:
https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/install-onnx.html
But how to get it integrated into piper?
@eliranwong
From what I know, you need to modify and rebuild Piper, setting the execution-providers to your chosen ones.
Also, you need to provide onnxruntime
build with your providers.
Best Musharraf
Appreciate your reply and help. May I ask for more information about:
Do you mean I need to manually edit this line:
@eliranwong
ROCM
, you need to modify Piper's C++ code to set it manually in the source.onnxruntime.so
built with ROCM
EP, if pre-built binaries are not provided by Microsoft, you need to build it yourself.So far, below is the easiest way that I found:
# pre-requisites
pip install -U pip
pip install cmake onnx
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
# Install ONNXRuntime from source
git clone --recursive https://github.com/ROCmSoftwarePlatform/onnxruntime.git
cd onnxruntime
git checkout rocm6.0_internal_testing
./build.sh --config Release --build_wheel --update --build --parallel --cmake_extra_defines ONNXRUNTIME_VERSION=$(cat ./VERSION_NUMBER) --use_rocm --rocm_home=/opt/rocm
pip install build/Linux/Release/dist/*
to:
providers=["ROCMExecutionProvider"]
I am open to better solution.
I would appreciate if the author of piper and support it directly, so that I don't need to manually edit the line.
Many thanks.
I read piper currently support --cuda argument. I would suggest to add --rocm argument to make piper better.
Does piper support AMD GPU acceleration with rocm?