CudaWindows

class lib.system.ml_libs.CudaWindows

Bases: _Cuda

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

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 Windows
`get_version`()	Attempt to get the default Cuda version from the Environment Variable
`get_versions`()	Attempt to detect all installed Cuda versions on Windows systems from the registry
`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 Windows

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

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

Attempt to get the default Cuda version from the Environment Variable

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

Attempt to detect all installed Cuda versions on Windows systems from the registry

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

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