diff options
Diffstat (limited to 'swr2_asr/__main__.py')
| -rw-r--r-- | swr2_asr/__main__.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/swr2_asr/__main__.py b/swr2_asr/__main__.py new file mode 100644 index 0000000..be294fb --- /dev/null +++ b/swr2_asr/__main__.py @@ -0,0 +1,12 @@ +"""Main entrypoint for swr2-asr.""" +import torch +import torchaudio + +if __name__ == "__main__": + # test if GPU is available + print("GPU available: ", torch.cuda.is_available()) + + # test if torchaudio is installed correctly + print("torchaudio version: ", torchaudio.__version__) + print("torchaudio backend: ", torchaudio.get_audio_backend()) + print("torchaudio info: ", torchaudio.get_audio_backend()) |