diff --git a/easyocr/recognition.py b/easyocr/recognition.py
index 530ef9517e2..7023b84e7c1 100644
--- a/easyocr/recognition.py
+++ b/easyocr/recognition.py
@@ -198,9 +198,10 @@ def get_text(character, imgH, imgW, recognizer, converter, image_list,\
     img_list = [item[1] for item in image_list]
     AlignCollate_normal = AlignCollate(imgH=imgH, imgW=imgW, keep_ratio_with_pad=True)
     test_data = ListDataset(img_list)
+    pin_memory = False if device == 'cpu' else True
     test_loader = torch.utils.data.DataLoader(
         test_data, batch_size=batch_size, shuffle=False,
-        num_workers=int(workers), collate_fn=AlignCollate_normal, pin_memory=True)
+        num_workers=int(workers), collate_fn=AlignCollate_normal, pin_memory=pin_memory)
 
     # predict first round
     result1 = recognizer_predict(recognizer, converter, test_loader,batch_max_length,\
@@ -214,7 +215,7 @@ def get_text(character, imgH, imgW, recognizer, converter, image_list,\
         test_data = ListDataset(img_list2)
         test_loader = torch.utils.data.DataLoader(
                         test_data, batch_size=batch_size, shuffle=False,
-                        num_workers=int(workers), collate_fn=AlignCollate_contrast, pin_memory=True)
+                        num_workers=int(workers), collate_fn=AlignCollate_contrast, pin_memory=pin_memory)
         result2 = recognizer_predict(recognizer, converter, test_loader, batch_max_length,\
                                      ignore_idx, char_group_idx, decoder, beamWidth, device = device)