specialized boy HarshaSatyavardhan
🎯
HarshaSatyavardhan
/ install_PyG.py
Created
December 26, 2023 08:55
— forked from ameya98/install_PyG.py
Google Colab: PyTorch Geometric Installation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Add this in a Google Colab cell to install the correct version of Pytorch Geometric. | |
| 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): |
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.