Skip to content

Instantly share code, notes, and snippets.

@vgthengane

vgthengane/pyg_install

Created December 9, 2021 14:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vgthengane/b3bbf1db55e15304090779319ebb1554 to your computer and use it in GitHub Desktop.
Save vgthengane/b3bbf1db55e15304090779319ebb1554 to your computer and use it in GitHub Desktop.
Download ZIP
Automatic PyTorch Geometric installation.
Raw
pyg_install
import torch
def format_pytorch_version(version):
return version.split('+')[0]
TORCH_version = torch.__version__
TORCH = format_pytorch_version(TORCH_version)
def format_cuda_version(version):
return 'cu' + version.replace('.', '')
CUDA_version = torch.version.cuda
CUDA = format_cuda_version(CUDA_version)
!pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-geometric
@richasinghai
Copy link

how to remove missing value for training so please provide me code for this data set.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment