mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-27 23:47:46 +00:00
you-n-g
c707a40073
* scen * scen2 * app * fix * Simplify workflow * We can share more code in new scenarios * rename model to rd loop * Optimize data path * Update rdagent/app/data_mining/model.py * Add TODO * Support GPU * gpu --------- Co-authored-by: SH-Src <suhan.c@outlook.com>
26 lines
746 B
Docker
26 lines
746 B
Docker
FROM pytorch/pytorch:latest
|
|
# 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/*
|
|
|
|
WORKDIR /workspace
|
|
|
|
RUN python -m pip install numpy
|
|
RUN python -m pip install --upgrade cython
|
|
# RUN python -m pip install -e .
|
|
|
|
RUN python -m pip install pandas
|
|
# RUN pip install pyg_lib torch_scatter torch_sparse torch_cluster -f https://data.pyg.org/whl/torch-2.3.0%2Bcu121.html
|
|
RUN pip install torch_geometric
|
|
RUN pip install ogb
|
|
RUN pip install networkx
|
|
RUN pip install scikit-learn
|
|
RUN pip install catboost
|
|
RUN pip install xgboost
|
|
RUN pip install sparse
|