2024-01-10 10:32:38 +08:00
|
|
|
import torch_geometric
|
|
|
|
|
import torch
|
|
|
|
|
if __name__ == '__main__':
|
2024-04-29 17:31:04 +08:00
|
|
|
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())
|
2024-01-10 10:32:38 +08:00
|
|
|
print(torch.cuda.nccl.version())
|
