github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-01-09 14:48:06 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
b897bddf38083ae363f5b7e09654a5380939fd10
ROCm/docs/reference/framework_compatibility/framework_compatibility.md
Sam Wu b897bddf38 Linkcheck and prepare alpha (#2078)
2023-04-24 11:25:31 -06:00

1.1 KiB
Raw Blame History

Framework Compatibility

The ROCm release supports the most recent and two prior releases of PyTorch and TensorFlow.

Legends:

Blue: Shows compatibility tested versions

Gray: Not tested

With PyTorch

With TensorFlow

Supported Frameworks

This section contains the latest release notes for each framework compatible with ROCm™ and Deep Learning (DL) applications.

The ROCm 5.4 platform supports the following frameworks:

• PyTorch v1.12.1

• MAGMA v2.5.4

• TensorFlow v2.10.0

PyTorch

For the latest release of PyTorch, refer to https://github.com/pytorch/pytorch/releases/

MAGMA

For the latest release of MAGMA, refer to https://icl.utk.edu/magma/index.html

TensorFlow

For the latest release of TensorFlow, refer to https://github.com/tensorflow/tensorflow/releases

Reference in New Issue View Git Blame Copy Permalink