跳到主要内容

版本发布信息

TensorFlow-MUSA v1.5.0

镜像地址及对应 MUSA 版本

针对不同的 MUSA SDK 和 Linux Driver 版本,提供以下两个镜像版本。镜像内已预置 MUSA SDK、TensorFlow 2.15.1 及编译好的 tensorflow_musa_extension,无需手动编译。

镜像内置 MUSA SDK依赖 Linux Driver推荐
registry.mthreads.com/presale/devtech/tensorflow_musa:5.1.0_202606235.1.05.1.0
registry.mthreads.com/presale/devtech/tensorflow_musa:4.3.5_202606244.3.53.3.5-

TensorFlow 版本兼容

TensorFlow-MUSA 开源项目兼容 TensorFlow 2.6.1TensorFlow 2.15.1,上述镜像并不预置 TensorFlow 2.6.1

如用户需要在 MUSA 环境使用 TensorFlow 2.6.1,参考 TensorFlow MUSA ExtensionREADME 自行构建并安装。

功能特性

  • 支持 30+ 搜推模型的训练和推理;

  • 支持 1000+ TensorFlow 算子;

  • 通过算子融合和图优化提升训推性能。