Updated Date: SEP 1, 2021

Install Pytorch

Follow the Pytorch tutorial, choose your verison. An example goes like this:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

Install Geometric

Follow the Pytorch Geometric tutorial, choose your version. An example goes like this:

conda install pytorch-geometric -c rusty1s -c conda-forge

Test installation

import torch
from torch_geometric.data import Data