feat: adding the docker file

This commit is contained in:
sohaibanwaar
2023-05-02 13:28:20 +05:00
parent 5b85c5433f
commit a693a779fa
4 changed files with 122 additions and 2 deletions

View File

@ -77,7 +77,16 @@ $ pip install -e .
You may also need to install ffmpeg, rust etc. Follow openAI instructions here https://github.com/openai/whisper#setup.
### Docker
Alternatively, you can use the docker image provided in this repo. To build the image, run the following command from the root of this repo:
1. In this image you can find jupyter notebook where you can easily run and debug the code.
```bash
docker build -t whisperx .
```
2. To run the image, run the following command:
```bash
docker run --gpus=all -it -v <local code dir>:/workspace -p 8888:8888 whisperx
```
### Setup not working???
Safest to use install pytorch as follows (for gpu)