mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-04 19:44:23 +08:00
12 lines
255 B
Python
12 lines
255 B
Python
import os
|
|
|
|
import torch
|
|
|
|
cmake_prefix_path = getattr(
|
|
torch.utils,
|
|
"cmake_prefix_path",
|
|
os.path.join(os.path.dirname(torch.__file__), "share", "cmake"),
|
|
)
|
|
version = torch.__version__.split("+")[0]
|
|
print(";".join([cmake_prefix_path, version]))
|