An 开源 机器学习 面向所有人的框架
[!IMPORTANT] 截至 2025 年 7 月 31 日,本项目已被淘汰。nvidia-tensorflow 不再维护。NVIDIA 建议新项目采用 JAX Toolbox。
NVIDIA 创建了这个项目,以支持使用 TensorFlow 1.x 的 NVIDIA GPU 用户的新硬件和改进的库。随着 TensorFlow 2.0 的发布,Google 宣布,在 2019 年 10 月 14 日发布 TF 1.15 之后,将不再为 TF 1.x 分支提供新的主要版本。NVIDIA 正在与 Google 和社区合作,通过添加对新硬件和库的支持来改进 TensorFlow 2.x。然而,有大量的 NVIDIA GPU 用户仍在其软件生态系统中使用 TensorFlow 1.x。本次发布将保持与上游 TensorFlow 1.15 版本的 API 兼容性。从今以后,本项目将被称为 nvidia-tensorflow。
链接到 Tensorflow README
nvidia-tensorflow 包含对 Linux 的 CPU 和 GPU 支持。
暂无开放 Issues,或尚未同步最近议题。