tensorflow-upstream
TensorFlow port for AMD GPUs via ROCm, enabling machine learning on Radeon hardware.
What is tensorflow-upstream?
TensorFlow ROCm port is a version of TensorFlow modified to support AMD GPUs through the ROCm platform. It solves the problem of TensorFlow's default limitation to NVIDIA CUDA hardware by enabling machine learning acceleration on AMD Radeon graphics cards. This allows users with AMD hardware to run TensorFlow workloads with GPU acceleration.
Machine learning researchers, data scientists, and developers who use AMD Radeon GPUs and want to leverage TensorFlow for their AI/ML projects. It's particularly valuable for those in AMD-centric hardware environments or seeking vendor diversity in their ML infrastructure.
Developers choose TensorFlow ROCm port because it provides official AMD GPU support within the TensorFlow ecosystem, eliminating the need for workarounds or custom builds. It offers a seamless experience with pre-built Docker containers and PyPI packages, making it easy to get started with AMD-accelerated machine learning.
Overview
TensorFlow ROCm port
Use Cases
Best For
- Running TensorFlow models on AMD Radeon GPUs
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
