From 4d49b710d80e9406e24fab9e32bdd3bb3c07d2f1 Mon Sep 17 00:00:00 2001
From: Yuta Hayashibe <yuta@hayashibe.jp>
Date: Sat, 24 Sep 2022 11:02:40 +0900
Subject: [PATCH] Add a setup step

---
 README.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/README.md b/README.md
index ba211ee..d28b008 100644
--- a/README.md
+++ b/README.md
@@ -16,6 +16,9 @@ git clone https://github.com/shirayu/whisper_streaming.git
 cd whisper_streaming
 poetry install --only main
 
+# If you use GPU with CUDA 11.6
+poetry run pip install 'torch==1.12.0+cu116' -f https://download.pytorch.org/whl/torch_stable.html
+
 # Run in English
 poetry run whisper_streaming --language en --model base -n 20
 ```