mirror of
https://github.com/SillyTavern/SillyTavern-Extras.git
synced 2026-04-29 19:01:20 +00:00
Create Dockerfile
This commit is contained in:
35
docker/Dockerfile
Normal file
35
docker/Dockerfile
Normal 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"]
|
||||
Reference in New Issue
Block a user