본문 바로가기

AI/CS231n

CS231n-Lec8. Deep Learning Software

반응형

CPU vs GPU

 

CUDA (NVIDIA only)

write C-like code that runs directly on the GPU 

highter-level APIs: cuDNN, cuBLAS, etc..

OpenCL 

similar to CUDA, but on anything 

slow 

 

cuDNN much faster than unoptimized CUDA 

 

Deep Learning Frameworks

 

Caffe / Caffe2

Theano/TensorFlow

Torch/PyTorch

 

The Point of deep learning frameworks

(1) Easily build big computational graphs 

(2) Easily compute gradients in computational graphs

(3) Run it all efficiently on GPU (warp cuDNN, cuBLAS, etc)

 

Computational Graphs 

Numpy 

problems:

can't run on GPU 

have to compute our own gradients 

반응형