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)

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.