Skip to content

luciusssss/CS224W-Colab

Repository files navigation

CS224W-Colab

Solutions for Stanford CS224W (Machine Learning with Graphs) Colab, Winter 2021

http://snap.stanford.edu/class/cs224w-2020/

http://web.stanford.edu/class/cs224w/

斯坦福大学 图机器学习课程 CS224W Colab答案

Problem with importing torch-geometric

If you encounter problems about importing torch-geometric on Colab, you can go to https://colab.research.google.com/drive/1h3-vJGRVloF5zStxL5I0rSy4ZUPNsjy8 to get the correct installing commands for the corresponding version of torch and cuda, such as

import os
import torch
os.environ['TORCH'] = torch.__version__
print(torch.__version__)

!pip install -q torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}.html
!pip install -q torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}.html
!pip install -q git+https://github.com/pyg-team/pytorch_geometric.git