2023 年 11 月 17 日 — TensorFlow 团队发布TensorFlow 2.15 已发布!此版本(以及 2.14)的亮点包括:为 Linux 上的 NVIDIA CUDA 库提供更简化的安装方法,Windows x64 和 x86 的 oneDNN CPU 性能优化,tf.function
类型的全面可用性,升级至 Clang 17.0.1,以及更多!有关完整的发行说明,请查看 这里。注意:有关新多后端 Keras 的发行更新将从 Keras 3.0 开始发布在 keras.io 上。有关更多信息,请查看 这里。
TensorFlow 2.15 已发布!此版本(以及 2.14)的亮点包括:为 Linux 上的 NVIDIA CUDA 库提供更简化的安装方法,Windows x64 和 x86 的 oneDNN CPU 性能优化,tf.function
类型的全面可用性,升级至 Clang 17.0.1,以及更多!有关完整的发行说明,请查看 这里。
tensorflow
pip 包现在为 Linux 提供了一种新的可选安装方法,该方法通过 pip 安装必要的 NVIDIA CUDA 库。只要系统上已经安装了 NVIDIA 驱动程序,您现在就可以运行 pip install tensorflow[and-cuda]
,在 Python 环境中安装 TensorFlow 的 NVIDIA CUDA 库依赖项。除了 NVIDIA 驱动程序之外,无需其他现有的 NVIDIA CUDA 包。在 TensorFlow 2.15 中,CUDA 已升级至 12.2 版本。
对于 Windows x64 和 x86 包,oneDNN 优化现在在 X86 CPU 上默认启用。可以通过在运行 TensorFlow 之前将环境变量 TF_ENABLE_ONEDNN_OPTS
设置为 1
(启用)或 0
(禁用)来启用或禁用这些优化。要恢复为默认设置,只需取消设置环境变量即可。
tf.function
类型现在已完全可用。
tf.types.experimental.TraceType
现在允许自定义tf.function
输入声明张量分解和类型转换支持。
- 介绍
tf.types.experimental.FunctionType
作为tf.function
可调用对象签名的综合表示。可以通过tf.function
和ConcreteFunctions
的function_type
属性访问它。有关更多详细信息,请参见tf.types.experimental.FunctionType
文档。
- 介绍
tf.types.experimental.AtomicFunction
作为在 Python 中执行 TF 计算的最快方法。可以通过ConcreteFunctions
的inference_fn
属性访问此功能。(不支持梯度。)有关如何调用和使用它,请参见tf.types.experimental.AtomicFunction
文档。
TensorFlow PIP 包现在使用 Clang 17 和 CUDA 12.2 构建,以提高基于 NVIDIA Hopper 的 GPU 的性能。展望未来,Clang 17 将成为 TensorFlow 的默认 C++ 编译器。建议您在从源代码构建 TensorFlow 时将编译器升级至 Clang 17。