pyg_lib
- cuda_version() int
Returns the CUDA version for which
pyg_lib
was compiled with.- Returns
The CUDA version.
- Return type
(int)
- get_home_dir() str
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)