Tensorflow 源码分析-GPU调用是如何实现的

1. Tensorflow GPU支持 Tensorflow 支持GPU进行运算,目前官方版本只支持NVIDIA的GPU,能够在tensorflow的官方上看到。Tensorflow 对GPU的运算的支持最小力度就是OP,也就是咱们常说的算子,下图提供了Tensorflow的一些常见算子,而每一个算子在Tensorflow上都会提供GPU的算法:关于OP的具体实现,在本篇博客中就不叙述了。 2.
相关文章
相关标签/搜索