mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-02 18:07:43 +00:00
Yuante Li
d6ce70b551
* fix model input shape bug and costeer_model bug * fix a bug * fix a bug in docker result extraction * a system-level optimization * add a filter of stdout * update * add stdout to model * model training_hyperparameters update * quant scenario * update some quant settings * llm choose action * Thompson Sampling Bandit for action choosing * refine both scens * add trace messages for quant scen * fix some bugs * fix some bugs * update * update * update * fix * fix * fix * update for merge * fix ci * fix some bugs * fix ci * fix ci * fix ci * fix ci * refactor * default qlib4rdagent local env downloading * fix ci * fix ci * fix a bug * fix ci * fix: align all prompts on template (#908) * use template to render all prompts * fix CI --------- Co-authored-by: Xu Yang <xuyang1@microsoft.com> * add fin_quant in cli * fix a bug * fix ci * fix some bugs * refactor * remove the columns in hypothesis if no value generated in this column * fix a bug * fix ci * fix conda env * add qlib gitignore * remove existed qlib folder & install torch in qlib conda * fix workspace ui in feedback * align model config in coder and runner in docker or conda * fix CI * fix CI --------- Co-authored-by: Xu Yang <peteryang@vip.qq.com> Co-authored-by: Xu Yang <xuyang1@microsoft.com>
25 lines
641 B
Docker
25 lines
641 B
Docker
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
|
|
|
|
# For GPU support, please choose the proper tag from https://hub.docker.com/r/pytorch/pytorch/tags
|
|
|
|
RUN apt-get clean && apt-get update && apt-get install -y \
|
|
curl \
|
|
vim \
|
|
git \
|
|
build-essential \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN git clone https://github.com/microsoft/qlib.git
|
|
|
|
WORKDIR /workspace/qlib
|
|
|
|
RUN git fetch && git reset 3e72593b8c985f01979bebcf646658002ac43b00 --hard
|
|
|
|
RUN python -m pip install --upgrade cython
|
|
RUN python -m pip install -e .
|
|
|
|
RUN pip install catboost
|
|
RUN pip install xgboost
|
|
RUN pip install scipy==1.11.4
|
|
RUN pip install tables
|