CV第八课 Deeplearning Frameworks (pytorch numpy tensorflow) 重点 !

选择使用深度学习框架的原因:

  1. 随着网络的更加复杂,深度学习的计算图我们越来越难画出来了,框架能简单的画出来

  2. 框架能帮助我们算梯度

  3. 框架帮助我们合理利用GPU 我们不用考虑写cuda代码或者cuDNN cuBLAS

        

使用numpy写forward and backward:

  1. 只能运行在CPU,无法运行在GPU

  2. 只能自己写backward梯度,分析Local gradient, upstream Gradient

Tensorflow:

  可以实现自动计算梯度,以及使用gpu/cpu :

  with tf.device('/gpu:0'):

Pytorch:

  

原文地址:https://www.cnblogs.com/ChevisZhang/p/12965396.html