NVIDIA GPU加速TensorFlow

NVIDIA开发的 apex PyTorch扩展,可以在Pytorch中进行 混合精度训练(Mixed Precision Training) 和分布式训练 (distributed training) : Introducing Apex: PyTorch Extension with Tools to Realize the Power of Tensor Cores

参考