构建 WholeGraph 容器#

要运行 WholeGraph 或从源代码构建 WholeGraph，首先需要设置环境。我们推荐使用 Docker 镜像。例如，要从 NGC pytorch 22.10 镜像构建 WholeGraph 基础镜像，可以按照 Dockerfile 操作

FROM nvcr.io/nvidia/pytorch:22.10-py3

RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y lsb-core software-properties-common wget libspdlog-dev

#RUN remove old cmake to update
RUN conda remove --force -y cmake
RUN rm -rf /usr/local/bin/cmake && rm -rf /usr/local/lib/cmake && rm -rf /usr/lib/cmake

RUN apt-key adv --fetch-keys https://apt.kitware.com/keys/kitware-archive-latest.asc && \
    export LSB_CODENAME=$(lsb_release -cs) && \
    apt-add-repository -y "deb https://apt.kitware.com/ubuntu/ ${LSB_CODENAME} main" && \
    apt update && apt install -y cmake

# update py for pytest
RUN pip3 install -U py
RUN pip3 install Cython setuputils3 scikit-build nanobind pytest-forked pytest

要运行 GNN 应用，您可能还需要 DGL 和/或 PyG 库来运行 GNN 层。您可以参考 DGL 或 PyG 例如，要安装 DGL，您可能需要添加

RUN pip3 install  dgl -f https://data.dgl.ai/wheels/torch-2.3/cu118/repo.html

本页