9 lines
262 B
Python
9 lines
262 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.nccl.is_available())
|
|
print(torch.cuda.nccl.version())
|