9 lines
265 B
Python
9 lines
265 B
Python
|
import torch
|
||
|
|
import torch_geometric
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print(f"Torch version: {torch.__version__}")
|
||
|
|
print(f"CUDA version: {torch.version.cuda}")
|
||
|
|
print(f"Cuda is available? {torch.cuda.is_available()}")
|
||
|
|
print(f"PyG version: {torch_geometric.__version__}")
|