import caffe
import caffe as GPUCaffeimport caffe as CPUCaffeCPUcaffe.set_mode_cpu()GPUcaffe.set_mode_gpu()