Prep for runpod docker

This commit is contained in:
Jaret Burkett
2024-08-03 12:41:06 -06:00
parent de2da96a81
commit 88acc28d7f
3 changed files with 24 additions and 2 deletions

21
docker/Dockerfile Normal file
View File

@@ -0,0 +1,21 @@
FROM runpod/base:0.6.2-cuda12.1.0
LABEL authors="jaret"
# Install dependencies
RUN apt-get update
WORKDIR /app
ARG CACHEBUST=1
RUN git clone https://github.com/ostris/ai-toolkit.git && \
cd ai-toolkit && \
git submodule update --init --recursive
WORKDIR /app/ai-toolkit
RUN ln -s /usr/bin/python3 /usr/bin/python
RUN python -m pip install -r requirements.txt
RUN apt-get install -y tmux nvtop htop
WORKDIR /
CMD ["/start.sh"]