AMD ROCm software is AMD's Open Source stack for GPU computation.
To learn more about ROCm, check out our Documentation, Examples, and Developer Hub.
If you have questions or need help, reach out to us on GitHub.
AMD ROCm software is AMD's Open Source stack for GPU computation.
To learn more about ROCm, check out our Documentation, Examples, and Developer Hub.
If you have questions or need help, reach out to us on GitHub.
Tensors and Dynamic neural networks in Python with strong GPU acceleration. To use PyTorch with latest ROCm releases, see: https://rocm.docs.amd.com/en/latest/rocm-for-ai/pytorch.html