Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
Dockerfile_ascend_910b 998 Bytes
Newer Older
litzh's avatar
Initial commit  
litzh committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31 

FROM quay.io/ascend/vllm-ascend:v0.11.0rc3

# Set envs
ENV PYTHONPATH=/workspace/LightX2V
ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/Ascend/ascend-toolkit/latest/lib64:/usr/local/Ascend/driver/lib64/driver/:/usr/local/python3.11.13/lib/python3.11/site-packages/torch_npu/lib/:/usr/local/python3.11.13/lib/python3.11/site-packages/torch/lib
ENV PATH=${PATH}:/usr/local/go/bin:/usr/local/python3.11.13/bin/

# Install deps
RUN apt-get update && apt-get install ffmpeg -y && \
    pip install --no-cache-dir \
        imageio \
        imageio-ffmpeg \
        ftfy \
        aiohttp \
        gguf \
        loguru \
        accelerate \
        diffusers && \
    pip install --no-cache-dir torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cpu

# Copy files
COPY app app
COPY assets assets
COPY configs configs
COPY lightx2v lightx2v
COPY lightx2v_kernel lightx2v_kernel
COPY lightx2v_platform lightx2v_platform
COPY scripts scripts
COPY test_cases test_cases
COPY tools tools