mirror of
https://github.com/shirayu/whispering.git
synced 2024-06-13 02:39:23 +00:00
add Dockerfile
This commit is contained in:
parent
74ed23a9e3
commit
efa01189e7
19
Dockerfile
Normal file
19
Dockerfile
Normal file
|
@ -0,0 +1,19 @@
|
|||
FROM nvidia/cuda:11.8.0-base-ubuntu22.04
|
||||
|
||||
|
||||
|
||||
# install python
|
||||
RUN apt-get update && apt-get install -y python3.8 python3-pip
|
||||
|
||||
# install portaudio
|
||||
RUN apt -y install portaudio19-dev
|
||||
|
||||
# torch
|
||||
RUN pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
|
||||
|
||||
# Copy entire repo and install
|
||||
COPY . /app
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
RUN pip install .
|
Loading…
Reference in a new issue