how to use GPU in example like 'python driver.py'

karpathy / neuraltalk

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.

5.4k stars 1.32k forks source link

Closed andyyuan78 closed 9 years ago

andyyuan78 commented 9 years ago

it seems it doesn't use GPU by default

karpathy commented 9 years ago

it doesn't use gpu at all

andyyuan78 commented 9 years ago

that's bad news!

EricZeiberg commented 9 years ago

You can use GPU for caffe predictions, as long as you have the CUDA software installed.