FROM pytorch/pytorch:1.7.0-cuda11.0-cudnn8-runtime
ENV TORCH=1.7.0
ENV CUDA=cu110

#ARG HTTP_PROXY
#ENV HTTP_PROXY $HTTP_PROXY
#ENV HTTPS_PROXY $HTTP_PROXY
#ENV FTP_PROXY $HTTP_PROXY
#ENV http_proxy $HTTP_PROXY
#ENV https_proxy $HTTP_PROXY

RUN pip install --force torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
RUN pip install --force torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
RUN pip install --force torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
RUN pip install --force torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
RUN pip install torch-geometric

WORKDIR /work

COPY ./ /work/

RUN cd /work;pip install -r requirements.txt

ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
