Spaces:
Sleeping
Sleeping
File size: 886 Bytes
6a62ffb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
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 torchvision torchaudio pytorch-cuda=11.7 git -c pytorch -c nvidia -c conda-forge
WORKDIR /sillytavern-extras/
COPY . .
ARG REQUIREMENTS
RUN pip install -r $REQUIREMENTS
ARG MODULES
CMD ["python","server.py","--enable-modules=$MODULES"]
|