mirror of
https://github.com/SillyTavern/SillyTavern-Extras.git
synced 2026-02-08 07:20:13 +00:00
36 lines
890 B
Docker
36 lines
890 B
Docker
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"]
|