Docker Installation¶
PyTorch provides a Docker image with PyTorch preinstalled. Using this image, the Dockerfile below creates a Docker image with PathpyG installed. PathpyG requires PyTorch Geometric version 2.5.0 or above.
FROM pytorch/pytorch:2.8.0-cuda12.9-cudnn9-runtime
WORKDIR /workspaces/pathpyG
RUN apt-get update
RUN apt-get -y install git
RUN pip install "torch_geometric>=2.5.0"
RUN pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.8.0+cu129.html
RUN pip install git+https://github.com/pathpy/pathpyG.git
FROM python:3.12-slim
WORKDIR /workspaces/pathpyG
RUN apt-get update
RUN apt-get -y install git
RUN pip install torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu # CPU only
RUN pip install "torch_geometric>=2.5.0"
RUN pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.8.0+cpu.html # CPU only
RUN pip install git+https://github.com/pathpy/pathpyG.git