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
Build CuPy against ROCm? #11
Comments
ROCM compiler stack is on conda-forge, conda-forge/staged-recipes#10123 |
Cool, thanks for the info @isuruf! Any estimation for when the ROCm support will be up and running? |
According to cupy docs, we need |
A quick update: |
Due to 1. lack of interest from AMD, 2. lack of bandwidth, 3. potential conflict of interests, I am closing this issue as "won't fix". |
This issue won't be resolvable in the foreseeable future (at least I don't think so -- lots of extra work and time investment are needed), but it's still worth being kept track.
CuPy recently adds experimental support for ROCm (targeting AMD GPUs): https://docs-cupy.chainer.org/en/latest/install_rocm.html.
Now, there's a
roctools
from Numba: https://anaconda.org/numba/roctools, but based on the Numba issue tracker and the download times, I doubt it's been extensively tested. Moreover, it doesn't provide the full ROCm software stack, just some LLVM bindings. Users still need to install the ROCm environment themselves: http://numba.pydata.org/numba-doc/latest/roc/overview.html#installation.For CuPy + ROCm to work in conda, I imagine we need the following things to work:
cudatoolkit
(canroctools
be reused here?)Disclaimer: I know nothing about ROCm. 🙂
The text was updated successfully, but these errors were encountered: