11 lines
353 B
Docker
11 lines
353 B
Docker
|
FROM pytorch/pytorch:latest
|
||
|
|
|
||
|
|
RUN apt update && \
|
||
|
|
apt install -y git curl unzip vim && \
|
||
|
|
pip install git+https://github.com/derfred/lightning.git@waifu-1.6.0#egg=pytorch-lightning
|
||
|
|
RUN mkdir /waifu
|
||
|
|
COPY . /waifu/
|
||
|
|
WORKDIR /waifu
|
||
|
|
RUN grep -v pytorch-lightning requirements.txt > requirements-waifu.txt && \
|
||
|
|
pip install -r requirements-waifu.txt
|