mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
31 lines
584 B
Docker
31 lines
584 B
Docker
FROM runpod/base:0.6.2-cuda12.2.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
|
|
|
|
RUN pip install jupyterlab
|
|
|
|
# mask workspace
|
|
RUN mkdir /workspace
|
|
|
|
|
|
# symlink app to workspace
|
|
RUN ln -s /app/ai-toolkit /workspace/ai-toolkit
|
|
|
|
WORKDIR /
|
|
CMD ["/start.sh"] |