imcomking/pytorch_geometric repository overview

FROM registry.navercorp.com/ami/pytorch:1.1-py3.6.2-cuda9.2

RUN apt-get update && apt-get install -y --no-install-recommends
autoconf automake libtool curl make g++ unzip

RUN rm -rf /var/lib/apt/lists/*

RUN pip install torch-geometric
torch-scatter
torch-sparse
torch-cluster
torch-spline-conv
cloudpickle

WORKDIR /workspace