浏览代码

updates to installation

Kolja Beigel 1 年之前
父节点
当前提交
9c012f8129
共有 3 个文件被更改,包括 24 次插入21 次删除
  1. 18 16
      README.md
  2. 1 5
      install_with_gpu_support.bat
  3. 5 0
      requirements-gpu.txt

+ 18 - 16
README.md

@@ -59,29 +59,31 @@ print(recorder.text())
 pip install RealtimeSTT
 ```
 
-## GPU Support
+This will install all the necessary dependencies, including a CPU support only version of PyTorch.
 
-To significantly improve transcription speed, especially in real-time applications, I **strongly recommend** utilizing GPU acceleration via CUDA. By default, the transcription is performed on the CPU. 
+### GPU Support with CUDA (recommended)
+
+Additional steps are needed for a GPU-optimized installation. These steps are recommended for those who require better performance and have a compatible NVIDIA GPU.
+
+If you plan to use RealtimeSTT with GPU support via CUDA, please follow these steps:
 
 1. **Install NVIDIA CUDA Toolkit 11.8**:
-	- Visit [NVIDIA CUDA Toolkit Archive](https://developer.nvidia.com/cuda-11-8-0-download-archive).
-	- Select version 11.
-	- Download and install the software.
+    - Visit [NVIDIA CUDA Toolkit Archive](https://developer.nvidia.com/cuda-11-8-0-download-archive).
+    - Select version 11.
+    - Download and install the software.
 
 2. **Install NVIDIA cuDNN 8.7.0 for CUDA 11.x**:
-	- Visit [NVIDIA cuDNN Archive](https://developer.nvidia.com/rdp/cudnn-archive).
-	- Click on "Download cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x".
-	- Download and install the software.
-	
-3. **Reconfigure PyTorch for CUDA**:
-	- If you have PyTorch CPU version installed, remove it: `pip uninstall torch` (CPU gets installed with the pip install RealtimeSTT command)
-	- Install PyTorch again with CUDA support: `pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118`.
+    - Visit [NVIDIA cuDNN Archive](https://developer.nvidia.com/rdp/cudnn-archive).
+    - Click on "Download cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x".
+    - Download and install the software.
 
-Note: To check if your NVIDIA GPU supports CUDA, visit the [official CUDA GPUs list](https://developer.nvidia.com/cuda-gpus).
-
-## Quick Start
+3. **Install PyTorch with CUDA support**:
+    ```bash
+    pip uninstall torch
+    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+    ```
 
-Here's a basic usage example:
+**Note**: To check if your NVIDIA GPU supports CUDA, visit the [official CUDA GPUs list](https://developer.nvidia.com/cuda-gpus).
 
 ### Automatic Recording
 

+ 1 - 5
install_with_gpu_support.bat

@@ -1,7 +1,3 @@
 pip uninstall torch
 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
-pip install PyAudio>=0.2.13
-pip install faster-whisper>=0.7.1
-pip install pvporcupine==1.9.5
-pip install webrtcvad>=2.0.10
-pip install halo>=0.0.31
+pip install -r requirements-gpu.txt

+ 5 - 0
requirements-gpu.txt

@@ -0,0 +1,5 @@
+PyAudio>=0.2.13
+faster-whisper>=0.7.1
+pvporcupine==1.9.5
+webrtcvad>=2.0.10
+halo>=0.0.31