import torch
import torch_geometric

def main():
    print(f"Torch version: {torch.__version__}")
    print(f"CUDA version: {torch.version.cuda}")
    print(f"Cuda is available? {torch.cuda.is_available()}")
    print(f"PyG version: {torch_geometric.__version__}")