10 lines
296 B
Python
10 lines
296 B
Python
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())
|