tensorflow-upstream

The tensorflow-upstream repository provides a ROCm port of TensorFlow for the project.

github.com/ROCmSoftwarePlatform/tensorflow-ups…
tensorflow-upstream homepage screenshot

About tensorflow-upstream

tensorflow-upstream is a GitHub repository containing the ROCm port of TensorFlow, the open source end-to-end machine learning platform. It packages the framework for use with the ROCm software stack and provides ROCm-specific setup documentation in its README.

The repository offers two main ways to get the ROCm build: a prebuilt Docker image (rocm/tensorflow:latest) published on Docker Hub, and tensorflow-rocm wheel packages on PyPI that can be installed with pip. Installation notes cover ROCm dependencies such as rocm-libs and rccl via apt, and the project uses GitHub Issues for bug tracking, with general questions directed to the TensorFlow Forum or Stack Overflow. Pricing information is not stated.

Key features

  • ROCm-targeted build of the TensorFlow machine learning framework
  • Docker image rocm/tensorflow:latest available on Docker Hub
  • tensorflow-rocm wheel packages on PyPI installable via pip
  • ROCm-specific README documenting port setup
  • Stable Python and C++ TensorFlow APIs
  • Bug tracking through GitHub Issues
Best for: Developers and researchers who want to run TensorFlow on a ROCm software stack rather than a standard CUDA-based setup.
Good for
  • Open source, with contribution guidelines, code of conduct and security files included
  • Provides both a ready-made Docker image and PyPI wheel packages for installation
  • Ships ROCm-specific documentation alongside the port
Watch out for
  • Users must set up a ROCm stack separately before installing the port
  • The documented apt-based dependency installation example targets Ubuntu
  • APIs for languages other than Python and C++ are described as not guaranteed to be backward compatible

tensorflow-upstream at a glance

Type
Open-source project
Pricing
Not confirmed. Check the site
Open source
Yes · Source code
Language
English
Last checked
Oct 6, 2026 · Online when checked
Listed
Oct 5, 2026

tensorflow-upstream alternatives

Tools and sites you could use instead, ranked by how closely they match.

All 24 alternatives

Popular in AI Infrastructure & MLOps

Well-known listings from the same topic.

All 341

tensorflow-upstream FAQ

How do I install the ROCm port of TensorFlow?
Install ROCm dependencies such as rocm-libs and rccl, then run pip3 install --user tensorflow-rocm --upgrade to get the wheel package from PyPI.
Is there a prebuilt container for this port?
Yes. A docker container, rocm/tensorflow:latest, is available on Docker Hub and is shown running with access to ROCm devices.
Where should bugs and questions be reported?
GitHub Issues is used for tracking requests and bugs; general questions go to the TensorFlow Forum, and specific questions to Stack Overflow.
Is something wrong or out of date?

Tell us and an editor will check it.

Own this website?

Claim the listing and suggest updates yourself. An editor checks every claim first.

Claim this listing