New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support for Radeon GPUs? #3115
Comments
I've tried using the PyTorch ROCm version from here However I get the following error when running my project:
|
Has DJL ever used Radeon GPUs? |
We don't support ROCm, you can try to build PyTorch JNI for ROCm by yourself. See: https://github.com/deepjavalibrary/djl/blob/master/engines/pytorch/pytorch-native/build.sh#L26 |
You actually can use DJL with ROCm using OnnxRuntime engine, see: https://github.com/deepjavalibrary/djl/blob/master/engines/onnxruntime/onnxruntime-engine/src/main/java/ai/djl/onnxruntime/engine/OrtModel.java#L212-L213 |
That sounds great, do you have any object detection inference examples using OnnxRuntime? |
Description
Support Radeon GPUs to be used for accelerating inferencing and training
Will this change the current api? How?
No
Who will benefit from this enhancement?
All users of Radeon GPUs
The text was updated successfully, but these errors were encountered: