I bought today a new graphic card and i want to train my model on GPU. My problem is, it take minutes to do something and instead of training, it stops. On CPU it works grea