dmlc--dgl
Hongzhi (Steve), Chen
98325b1097
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
12 行
255 B
Python
12 行
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]))
|