9 lines
427 B
Docker
9 lines
427 B
Docker
FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
|
|
|
|
WORKDIR /workspace
|
|
|
|
RUN pip install pytorch-lightning opencv-python-headless imageio[ffmpeg] einops datasets==3.2.0 simplejson diffusers==0.34.0 open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict -i https://nexus.4pd.io/repository/pypi-all/simple
|
|
RUN pip install transformers accelerate -i https://nexus.4pd.io/repository/pypi-all/simple
|
|
|
|
COPY . /workspace/
|