add Dockerfile

This commit is contained in:
ricwo 2022-11-08 12:15:19 +01:00
parent 74ed23a9e3
commit efa01189e7

19
Dockerfile Normal file
View 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 .