28 lines
822 B
Docker
28 lines
822 B
Docker
# syntax=docker/dockerfile:1.4
|
||
|
||
FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel AS tslib
|
||
|
||
WORKDIR /workspace
|
||
|
||
ARG http_proxy
|
||
ARG https_proxy
|
||
|
||
ENV http_proxy=${http_proxy}
|
||
ENV https_proxy=${https_proxy}
|
||
ENV PYTHONPATH=/workspace/Time-Series-Library:$PYTHONPATH
|
||
|
||
COPY requirements.txt .
|
||
RUN --mount=type=cache,target=/root/.cache/pip \
|
||
pip install -r requirements.txt
|
||
|
||
# mamba-ssm (cxx11abiFALSE) （Time-Series-Library/models/Mamba.py）
|
||
RUN --mount=type=cache,target=/root/.cache/pip \
|
||
pip install https://github.com/state-spaces/mamba/releases/download/v2.2.6.post3/mamba_ssm-2.2.6.post3+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
|
||
|
||
# uni2ts (--no-deps)（Time-Series-Library/models/Moirai.py）
|
||
RUN --mount=type=cache,target=/root/.cache/pip \
|
||
pip install uni2ts --no-deps
|
||
|
||
COPY . .
|
||
|
||
CMD ["bash"] |