Spaces:
Sleeping
Sleeping
File size: 2,993 Bytes
e75a247 44b7575 e75a247 7bd4c44 e75a247 44b7575 e75a247 44b7575 e75a247 96e40d3 e75a247 44b7575 |
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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
# gkrz/lgatr:v3
# docker build -t gkrz/lgatr:v4 .
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
WORKDIR /app
COPY . /app
SHELL ["/bin/bash", "-c"]
USER root
RUN apt update && \
DEBIAN_FRONTEND=noninteractive apt install --yes --no-install-recommends \
build-essential \
cmake \
ffmpeg \
git \
python-is-python3 \
python3-dev \
python3-pip \
&& \
rm -rf /var/lib/apt/lists/*
RUN python3.10 --version
RUN python3 --version
RUN python --version
RUN python3 -m pip install --no-cache-dir --upgrade pip
#python3 -m pip install --no-cache-dir --upgrade --requirement requirements.txt
RUN python3 -m pip install numba==0.58.1
# packages without conda
# RUN python3 -m pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
RUN python3 -m pip install torch torchvision torchaudio
RUN python3 -m pip install torch_geometric
RUN python3 -m pip install torch_scatter torch_sparse torch_cluster torch_spline_conv
RUN python3 -m pip install pytorch-lightning yacs torchmetrics
RUN python3 -m pip install performer-pytorch
RUN python3 -m pip install tensorboardX
RUN python3 -m pip install ogb
RUN python3 -m pip install wandb
RUN python3 -m pip install seaborn
RUN python3 -m pip install dgl
RUN python3 -m pip install numpy
RUN python3 -m pip install scipy
RUN python3 -m pip install pandas
RUN python3 -m pip install scikit-learn
RUN python3 -m pip install matplotlib
RUN python3 -m pip install tqdm
RUN python3 -m pip install PyYAML
RUN python3 -m pip install awkward0
RUN python3 -m pip install uproot
RUN python3 -m pip install awkward
RUN python3 -m pip install vector
RUN python3 -m pip install lz4
RUN python3 -m pip install xxhash
RUN python3 -m pip install tables
RUN python3 -m pip install tensorboard
RUN python3 -m pip install plotly
RUN python3 -m pip install xformers
RUN python3 -m pip install fastjet
RUN python3 -m pip install gradio
RUN python3 -m pip install huggingface_hub
RUN python3 -m pip install hdbscan
# remove pip cache
RUN python3 -m pip cache purge
# COPY docker/ext_packages /docker/ext_packages
# RUN python3 /docker/ext_packages/install_upstream_python_packages.py
RUN mkdir -p /opt/pepr
# Install GATr
#RUN cd /opt/pepr && git clone https://github.com/Qualcomm-AI-research/geometric-algebra-transformer.git geometric-algebra-transformer1
#RUN cd /opt/pepr/geometric-algebra-transformer1/ && python3 -m pip install .
# Install L-GATr - for some reason this only works if executed from the already-built container
RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
# Install torch_cmspepr
RUN cd /opt/pepr && git clone https://github.com/cms-pepr/pytorch_cmspepr
RUN cd /opt/pepr/pytorch_cmspepr/ && python3 -m pip install .
RUN cd /root
RUN ls
# entrypoint run app.py with python
ENTRYPOINT ["python", "app.py"]
|