けの〜のブログ

ガッキーでディープラーニングして教育界に革命を起こす

Deep Learning Frame work TensorFlowとPytorch

TensorFlowとPytorchについて書き留めたいと思う。

 

Deep learningでは大量の行列の計算を行う。

そのためGPUでの計算が実用上早いとされるため、Deep LearningにはGPUの設備が必要になってくる。

 

GPU上でプログラムを実行するにはGPU専用の言語があるため、NumpyなどをそのままGPUで稼働させることはできない。

 

そこでTensorFlowやPytorchと呼ばれるSoftWareを使用することが多い(たくさんある)

これらを使用することによってCPU,GPUどちらでも稼働可能で、傾きの計算も自動で行える。

 

TensorFlowは汎用性が高い

Pytorchは研究用に、Caffe2はapplicationへの応用が効きやすいようだ。