CudaLinux

class lib.system.ml_libs.CudaLinux

Bases: _Cuda

Find the location of system installed Cuda and cuDNN on Linux.

Methods Summary

`cudnn_version_from_header`(folder)	Attempt to detect the cuDNN version from the version header file
`get_cudnn_versions`()	Attempt to locate any installed cuDNN versions on Linux
`get_version`()	Attempt to locate the default Cuda version on Linux
`get_versions`()	Attempt to detect all installed Cuda versions on Linux systems
`version_from_version_file`(folder)	Attempt to get an installed Cuda version from its version.json file

Methods Documentation

cudnn_version_from_header(folder: str) → tuple[int, int, int] | None

Attempt to detect the cuDNN version from the version header file

Parameters:: folder (str) – The folder to check for the cuDNN header file
Return type:: The cuDNN version found from the given folder or None if not detected

get_cudnn_versions() → dict[tuple[int, int], tuple[int, int, int]]

Attempt to locate any installed cuDNN versions on Linux

Return type:: Detected cuDNN version for each installed Cuda. key (0, 0) denotes globally installed cudnn

get_version() → tuple[int, int] | None

Attempt to locate the default Cuda version on Linux

Checks, in order: update-alternatives, /usr/local/cuda, ldconfig, nvcc

get_versions() → dict[tuple[int, int], str]

Attempt to detect all installed Cuda versions on Linux systems

version_from_version_file(folder: str) → tuple[int, int] | None

Attempt to get an installed Cuda version from its version.json file

Parameters:: folder (str) – Full path to the folder to check for a version file
Return type:: The detected Cuda version or None if not detected