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://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html !pip install torch-sparse -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html !pip install torch-cluster -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html !pip install torch-spline-conv -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html !pip install torch-geometric !pip install networkx
machinelearningzuu/Machine-Learning-With-Graphs
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|