dmlc--dgl
Quan (Andy) Gan
9a7235faf2
* first commit * some thoughts * move around * more commit * more fixes * now it uses torch allocator * fix symbol export error * fix * fixes * test fix * add script * building separate library per version * fix for vs2019 * more fixes * fix on windows build * update jenkinsfile * auto copy built dlls for windows * lint and installation guide update * fix * specify conda environment * set environment for ci * fix * fix * fix * fix again * revert * fix cmake * fix * switch to using python interpreter path * remove scripts * debug * oops sorry * Update index.rst * Update index.rst * copies automatically, no need for this * do not print message if library not found * tiny fixes * debug on nightly * replace add_compile_definitions to make CMake 3.5 happy * fix linking to wrong lib for multiple pytorch envs * changed building strategy * fix nightly * fix windows * fix windows again * setup bugfix * address comments * change README
10 行
252 B
Python
10 行
252 B
Python
import torch
|
|
import os
|
|
|
|
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]))
|