pyg_lib
- cuda_version() int [source]
Returns the CUDA version for which
pyg_lib
was compiled with.- Returns:
The CUDA version.
- Return type:
(int)
- get_home_dir() str [source]
Gets the cache directory used for storing all
pyg-lib
data.If
set_home_dir()
is not called, the path is given by the environment variable$PYG_LIB_HOME
which defaults to"~/.cache/pyg_lib"
.- Returns:
The cache directory.
- Return type:
(str)