image-dreamer icon indicating copy to clipboard operation
image-dreamer copied to clipboard

Support for GPU not setup by default

Open paulnord opened this issue 8 years ago • 0 comments

If Caffe was built for the NVIDIA GPU, these lines will speed things up a LOT. Insert this code just after the "import caffe" line.

# If your GPU supports CUDA and Caffe was built with CUDA support,
# uncomment the following to run Caffe operations on the GPU.
caffe.set_mode_gpu()
caffe.set_device(0) # select GPU device if multiple devices exist

paulnord avatar Jan 05 '17 22:01 paulnord