Provide a newer dockerfile

[LukeLIN-web]

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

# Install PyG.
RUN CPATH=/usr/local/cuda/include:$CPATH && \
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH && \
    DYLD_LIBRARY_PATH=/usr/local/cuda/lib:$DYLD_LIBRARY_PATH

RUN pip install scipy==1.5.0

RUN pip install --no-index torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html && \
    pip install --no-index torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html && \
    pip install torch-geometric

WORKDIR /quiver
ADD . .
RUN pip install -v .

# Set the default command to python3.
CMD ["python3"]

It works on my host machine : Ubuntu 18.04 , NVIDIA-SMI 515.65.01 Driver Version: 515.65.01 CUDA Version: 11.7,Tesla V100-SXM2.

[LukeLIN-web]

dockerhub 1263810658