Skip to content

Commit

Permalink
docs: move cuDNN requirement to warning
Browse files Browse the repository at this point in the history
  • Loading branch information
winstxnhdw committed Feb 13, 2024
1 parent 940cf06 commit 8c0cc27
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@ A fast CPU-first video/audio transcriber for generating caption files with [Whis
## Requirements

- Python 3.11
- [NVIDIA cuDNN](https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-9.0.0.312_cuda11-archive.zip) (for Windows only)

## Usage (API)

Expand Down Expand Up @@ -43,6 +42,9 @@ pip install git+https://github.com/winstxnhdw/CapGen

You may also install `CapGen` with the necessary CUDA binaries.

> [!WARNING]\
> If you are on Windows, the NVIDIA cuDNN binary is not available on PyPI. You will have to install it manually from their official source [here](https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-9.0.0.312_cuda11-archive.zip).
```bash
pip install 'capgen[cuda] @ git+https://github.com/winstxnhdw/CapGen'
```
Expand Down

0 comments on commit 8c0cc27

Please sign in to comment.