Create Dockerfile

This commit is contained in:
ramblingcoder
2023-05-28 09:10:41 -05:00
committed by GitHub
parent 630f9a0054
commit 6512aa6fc3

35
docker/Dockerfile Normal file
View File

@@ -0,0 +1,35 @@
FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04
EXPOSE 5100
ENV PATH="/root/miniconda3/bin:${PATH}"
ARG PATH="/root/miniconda3/bin:${PATH}"
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
python3 python3-venv wget build-essential
RUN wget \
https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \
&& mkdir /root/.conda \
&& bash Miniconda3-latest-Linux-x86_64.sh -b \
&& rm -f Miniconda3-latest-Linux-x86_64.sh
RUN conda --version
RUN conda init
RUN conda create -n extras
RUN /bin/bash -c "source activate extras"
RUN conda install pytorch=2.0.0 torchvision=0.15.0 torchaudio=2.0.0 pytorch-cuda=11.7 git -c pytorch -c nvidia
WORKDIR /sillytavern-extras/
COPY . .
ARG REQUIREMENTS
RUN pip install -r $REQUIREMENTS
ARG MODULES
CMD ["python","server.py","--enable-modules=$MODULES"]