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.nccl.is_available()) print(torch.cuda.nccl.version())