Sucran
9/10/2017 - 11:51 AM

PyCaffe GPU mode

GPU mode

caffe.set_device(0)  # if we have multiple GPUs, pick the first one
caffe.set_mode_gpu()
net.forward()  # run once before timing to set up memory
%timeit net.forward() # time count