mxnet

更新时间:2022-09-13 14:43

MXNet 是亚马逊(Amazon)选择的深度学习库,并且也许是最优秀的库之一。

内容简介

MXNet 是亚马逊(Amazon)选择的深度学习库。它拥有类似于 Theano 和 TensorFlow 的数据流图,为多 GPU 配置提供了良好的配置,有着类似于 Lasagne 和 Blocks 更高级别的模型构建块,并且可以在你可以想象的任何硬件上运行(包括手机)。对 Python 的支持只是其冰山一角—MXNet 同样提供了对 R、Julia、C++、Scala、Matlab,和 Javascript 的接口。

1.0新特性

增强了sparse.dot运算符的性能。

MXNet自动设置OpenMP,以便在未设置NUM_OMP_THREADS时使用所有可用的CPU内核以最大限度地提高CPU利用率。

一元运算符和二元运算符可以避免在小型阵列上使用OpenMP,因为使用OpenMP实际上会损害多线程开销。

在CPU上显着提高了broadcast_add,broadcast_mul等的性能。

使用NCCL 2.1版或更新版本时,建议将环境变量NCCL_LAUNCH_MODE设置为PARALLEL。

MXNet可以加速任何数值计算[1]。神经网络中,每一层由一个线性函数和一个非线性变换组成。 在MXNet中,我们可以将其表示为

免责声明
隐私政策
用户协议
目录 22
0{{catalogNumber[index]}}. {{item.title}}
{{item.title}}