FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel

COPY requirements.txt .

RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html && \
    pip install --no-cache-dir torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.10.0+cu113.html && \
    pip install --no-cache-dir -r requirements.txt && \
    pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html && \
    #pip install --no-cache-dir -e . && \    
    echo 'Done'

