mirror of
https://github.com/ROCm/ROCm.git
synced 2026-01-09 14:48:06 -05:00
99 lines
4.6 KiB
ReStructuredText
99 lines
4.6 KiB
ReStructuredText
:orphan:
|
||
|
||
.. meta::
|
||
:description: Taichi compatibility
|
||
:keywords: GPU, Taichi, deep learning, framework compatibility
|
||
|
||
.. version-set:: rocm_version latest
|
||
|
||
*******************************************************************************
|
||
Taichi compatibility
|
||
*******************************************************************************
|
||
|
||
`Taichi <https://www.taichi-lang.org/>`_ is an open-source, imperative, and parallel
|
||
programming language designed for high-performance numerical computation.
|
||
Embedded in Python, it leverages just-in-time (JIT) compilation frameworks such as LLVM to accelerate
|
||
compute-intensive Python code by compiling it to native GPU or CPU instructions.
|
||
|
||
Taichi is widely used across various domains, including real-time physical simulation,
|
||
numerical computing, augmented reality, artificial intelligence, computer vision, robotics,
|
||
visual effects in film and gaming, and general-purpose computing.
|
||
|
||
Support overview
|
||
================================================================================
|
||
|
||
- The ROCm-supported version of Taichi is maintained in the official `https://github.com/ROCm/taichi
|
||
<https://github.com/ROCm/taichi>`__ repository, which differs from the
|
||
`https://github.com/taichi-dev/taichi <https://github.com/taichi-dev/taichi>`__ upstream repository.
|
||
|
||
- To get started and install Taichi on ROCm, use the prebuilt :ref:`Docker image <taichi-docker-compat>`,
|
||
which includes ROCm, Taichi, and all required dependencies.
|
||
|
||
- See the :doc:`ROCm Taichi installation guide <rocm-install-on-linux:install/3rd-party/taichi-install>`
|
||
for installation and setup instructions.
|
||
|
||
- You can also consult the upstream `Installation guide <https://github.com/taichi-dev/taichi>`__
|
||
for additional context.
|
||
|
||
Version support
|
||
--------------------------------------------------------------------------------
|
||
|
||
Taichi is supported on `ROCm 6.3.2 <https://repo.radeon.com/rocm/apt/6.3.2/>`__.
|
||
|
||
Supported devices
|
||
--------------------------------------------------------------------------------
|
||
|
||
- **Officially Supported**: AMD Instinct™ MI250X, MI210X (with the exception of Taichi’s GPU rendering system, CGUI)
|
||
- **Upcoming Support**: AMD Instinct™ MI300X
|
||
|
||
.. _taichi-recommendations:
|
||
|
||
Use cases and recommendations
|
||
================================================================================
|
||
|
||
* The `Accelerating Parallel Programming in Python with Taichi Lang on AMD GPUs
|
||
<https://rocm.blogs.amd.com/artificial-intelligence/taichi/README.html>`__
|
||
blog highlights Taichi as an open-source programming language designed for high-performance
|
||
numerical computation, particularly in domains like real-time physical simulation,
|
||
artificial intelligence, computer vision, robotics, and visual effects. Taichi
|
||
is embedded in Python and uses just-in-time (JIT) compilation frameworks like
|
||
LLVM to optimize execution on GPUs and CPUs. The blog emphasizes the versatility
|
||
of Taichi in enabling complex simulations and numerical algorithms, making
|
||
it ideal for developers working on compute-intensive tasks. Developers are
|
||
encouraged to follow recommended coding patterns and utilize Taichi decorators
|
||
for performance optimization, with examples available in the `https://github.com/ROCm/taichi_examples
|
||
<https://github.com/ROCm/taichi_examples>`_ repository. Prebuilt Docker images
|
||
integrating ROCm, PyTorch, and Taichi are provided for simplified installation
|
||
and deployment, making it easier to leverage Taichi for advanced computational workloads.
|
||
|
||
.. _taichi-docker-compat:
|
||
|
||
Docker image compatibility
|
||
================================================================================
|
||
|
||
.. |docker-icon| raw:: html
|
||
|
||
<i class="fab fa-docker"></i>
|
||
|
||
AMD validates and publishes ready-made `ROCm Taichi Docker images <https://hub.docker.com/r/rocm/taichi/tags>`_
|
||
with ROCm backends on Docker Hub. The following Docker image tag and associated inventories
|
||
represent the latest Taichi version from the official Docker Hub.
|
||
Click |docker-icon| to view the image on Docker Hub.
|
||
|
||
.. list-table::
|
||
:header-rows: 1
|
||
:class: docker-image-compatibility
|
||
|
||
* - Docker image
|
||
- ROCm
|
||
- Taichi
|
||
- Ubuntu
|
||
- Python
|
||
|
||
* - .. raw:: html
|
||
|
||
<a href="https://hub.docker.com/layers/rocm/taichi/taichi-1.8.0b1_rocm6.3.2_ubuntu22.04_py3.10.12/images/sha256-e016964a751e6a92199032d23e70fa3a564fff8555afe85cd718f8aa63f11fc6"><i class="fab fa-docker fa-lg"></i> rocm/taichi</a>
|
||
- `6.3.2 <https://repo.radeon.com/rocm/apt/6.3.2/>`_
|
||
- `1.8.0b1 <https://github.com/taichi-dev/taichi>`_
|
||
- 22.04
|
||
- `3.10.12 <https://www.python.org/downloads/release/python-31012/>`_ |