mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
Prep for runpod docker
This commit is contained in:
21
docker/Dockerfile
Normal file
21
docker/Dockerfile
Normal 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"]
|
||||
Reference in New Issue
Block a user