Meanwhile, I am doing feature extraction experiment on CPU and every time it freezes. At this moment, I have also a GPU and i want to consume GPU for feature extraction and