黄家有一堆基于cuda的库，提供几乎全过程的gpu加速
也是培育了多年了
https://developer.nvidia.com/gpu-accelerated-libraries#deep-learning
【 在 KM7 (楷魔) 的大作中提到: 】
: 主要还是Nvidia出的cuDNN比较流行吧，拿N家的那一套跑AI训练，自然A卡凑不上来
: CUDA搭配cuDNN，也算是N家比较宽的护城河了，AMD就拿不出这些
: 苹果现在ML compute也是类似cuDNN的东西
: ...................
--
FROM 73.63.245.*