import torch_geometric
import torch
if __name__ == '__main__':
    print(torch.__version__)
    print(torch.cuda.device_count())
    print(torch.cuda.get_device_name())
    print(torch.cuda.is_available())
    # print(torch.cuda.nccl.is_available())
    print(torch.cuda.nccl.version())