diff --git a/long_audio_transcribe.py b/long_audio_transcribe.py
index 1242aae..e2292bb 100644
--- a/long_audio_transcribe.py
+++ b/long_audio_transcribe.py
@@ -27,6 +27,7 @@ if __name__ == "__main__":
         lang2token = {
             'zh': "[ZH]",
         }
+    assert(torch.cuda.is_available()), "Please enable GPU in order to run Whisper!"
     model = whisper.load_model(args.whisper_size)
     speaker_annos = []
     for file in filelist:
diff --git a/short_audio_transcribe.py b/short_audio_transcribe.py
index 8a460de..04b23ef 100644
--- a/short_audio_transcribe.py
+++ b/short_audio_transcribe.py
@@ -2,6 +2,7 @@ import whisper
 import os
 import torchaudio
 import argparse
+import torch
 
 lang2token = {
             'zh': "[ZH]",
@@ -47,6 +48,7 @@ if __name__ == "__main__":
         lang2token = {
             'zh': "[ZH]",
         }
+    assert (torch.cuda.is_available()), "Please enable GPU in order to run Whisper!"
     model = whisper.load_model(args.whisper_size)
     parent_dir = "./custom_character_voice/"
     speaker_names = list(os.walk(parent_dir))[0][1]