2024-09-11 15:26:52 +08:00
|
|
|
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
|
2024-07-15 10:08:20 +00:00
|
|
|
|
|
|
|
|
# For GPU support, please choose the proper tag from https://hub.docker.com/r/pytorch/pytorch/tags
|
2024-07-15 08:28:34 +00:00
|
|
|
|
|
|
|
|
RUN apt-get clean && apt-get update && apt-get install -y \
|
|
|
|
|
curl \
|
|
|
|
|
vim \
|
|
|
|
|
git \
|
|
|
|
|
build-essential \
|
2025-10-10 08:29:22 +05:45
|
|
|
coreutils \
|
2024-07-15 08:28:34 +00:00
|
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
2025-04-23 18:48:02 +08:00
|
|
|
RUN git clone https://github.com/microsoft/qlib.git
|
2024-07-15 08:28:34 +00:00
|
|
|
|
|
|
|
|
WORKDIR /workspace/qlib
|
|
|
|
|
|
2026-02-13 10:50:46 +08:00
|
|
|
RUN git fetch && git reset 2fb9380b342556ddb50a4b24e4fe8655d548b2b8 --hard
|
2024-07-15 08:28:34 +00:00
|
|
|
|
|
|
|
|
RUN python -m pip install --upgrade cython
|
|
|
|
|
RUN python -m pip install -e .
|
|
|
|
|
|
|
|
|
|
RUN pip install catboost
|
2024-07-19 14:59:45 +08:00
|
|
|
RUN pip install xgboost
|
2025-03-20 11:39:19 +08:00
|
|
|
RUN pip install tables
|
