Allow GPU acceleration with AMD GPUs with ROCm #160
Allow GPU acceleration with AMD GPUs with ROCm #160CodingAndCodingAgain wants to merge 1 commit into
Conversation
docs: updated docs
But even if you install onnxruntime_rocm (Show)Use AMD official for rocm-rel-7.0.2 instead pip:
It goes with The situation won't change - Python still throws an error. When attempting to use MIGraphXExecutionProvider, the following message appears: But it still uses the CPU! After Any ideas? |
I was able to run piper with an AMD GPU by installing the "onnxruntime-rocm" and the appropriate "torch" package based on my GPU.
Without taking into account the installation of the packages, it didn't require to much setup to get it to work so I thought more people could benefit of it.
Importing the "torch" package before the "onnxruntime" package it's actually necessary since "onnxruntime-rocm" will crash without it.