pyg_lib

cuda_version() → int[source]

Returns the CUDA version for which pyg_lib was compiled with.

Returns:

int – The CUDA version.

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:

str – The cache directory.

set_home_dir(path: str)[source]

Sets the cache directory used for storing all pyg-lib data.

Parameters:

path (str) – The path to a local folder.