mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
26553d725b
* Add Tensorflow * WIP * WIP * minor fmt * PR feedbacks * fix missed inconsistent formatting * WIP WIP WIP WIP * minor formatting update tensorflow-rocm docker images to rocm6.3.1 fix urls * WIP * fix typo and update wordlist * fix tables not rendering * fix table headings * add period * update tf dockers * fix link * fix link * wording * update historical compat * fix tensile link --------- Co-authored-by: Mátyás Aradi <matyas@streamhpc.com> Co-authored-by: Istvan Kiss <neon60@gmail.com>
44 lines
2.0 KiB
ReStructuredText
44 lines
2.0 KiB
ReStructuredText
.. meta::
|
|
:description: How to install deep learning frameworks for ROCm
|
|
:keywords: deep learning, frameworks, ROCm, install, PyTorch, TensorFlow, JAX, MAGMA, DeepSpeed, ML, AI
|
|
|
|
********************************************
|
|
Installing deep learning frameworks for ROCm
|
|
********************************************
|
|
|
|
ROCm provides a comprehensive ecosystem for deep learning development, including
|
|
:ref:`libraries <artificial-intelligence-apis>` for optimized deep learning operations and ROCm-aware versions of popular
|
|
deep learning frameworks and libraries such as PyTorch, TensorFlow, and JAX. ROCm works closely with these
|
|
frameworks to ensure that framework-specific optimizations take advantage of AMD accelerator and GPU architectures.
|
|
|
|
The following guides provide information on compatibility and supported
|
|
features for these ROCm-enabled deep learning frameworks.
|
|
|
|
* :doc:`PyTorch compatibility <../compatibility/ml-compatibility/pytorch-compatibility>`
|
|
* :doc:`TensorFlow compatibility <../compatibility/ml-compatibility/tensorflow-compatibility>`
|
|
* :doc:`JAX compatibility <../compatibility/ml-compatibility/jax-compatibility>`
|
|
|
|
This chart steps through typical installation workflows for installing deep learning frameworks for ROCm.
|
|
|
|
.. image:: ../data/how-to/framework_install_2024_07_04.png
|
|
:alt: Flowchart for installing ROCm-aware machine learning frameworks
|
|
:align: center
|
|
|
|
See the installation instructions to get started.
|
|
|
|
* :doc:`PyTorch for ROCm <rocm-install-on-linux:install/3rd-party/pytorch-install>`
|
|
* :doc:`TensorFlow for ROCm <rocm-install-on-linux:install/3rd-party/tensorflow-install>`
|
|
* :doc:`JAX for ROCm <rocm-install-on-linux:install/3rd-party/jax-install>`
|
|
|
|
.. note::
|
|
|
|
For guidance on installing ROCm itself, refer to :doc:`ROCm installation for Linux <rocm-install-on-linux:index>`.
|
|
|
|
Learn how to use your ROCm deep learning environment for training, fine-tuning, inference, and performance optimization
|
|
through the following guides.
|
|
|
|
* :doc:`rocm-for-ai/index`
|
|
|
|
* :doc:`llm-fine-tuning-optimization/index`
|
|
|