roop
roop copied to clipboard
Did you get it working? Please post what you did under this issue.
Given how many os+software+hardware problems are there - it's not possible for me to provide installation steps or troubleshooting advice for every case.
I request people who got it working to post what steps they took here along with their relevant specs. I'll go first.
OS: Linux (not wsl/vm)
GPU: AMD 6900XT
Then installed torch and torchvision from here and finally installed onnxruntime-rocm as follows:
git clone https://github.com/microsoft/onnxruntime
cd onnxruntime
./build.sh --config Release --build_wheel --update --build --parallel --cmake_extra_defines CMAKE_PREFIX_PATH=/opt/rocm/lib/cmake ONNXRUNTIME_VERSION=$ONNXRUNTIME_VERSION onnxruntime_BUILD_UNIT_TESTS=off --use_rocm --rocm_home=/opt/rocm
pip install build/Linux/Release/dist/*.whl