cuda#

检测 CUDA 版本。

函数#

cuda_version()

尝试检测操作系统中存在的 CUDA 版本。

cached_cuda_version()

CUDA 检测系统的缓存版本。

conda_virtual_packages()

_cuda_driver_version_detector_target(queue)

尝试在操作系统中检测 CUDA 版本,在

cuda_version()#

尝试检测操作系统中存在的 CUDA 版本。

在 Windows 和 Linux 上,CUDA 库由 NVIDIA 驱动程序包安装,并且通常在标准库路径中找到,而不是在 CUDA SDK 中(CUDA SDK 对于运行 CUDA 应用程序是可选的)。

在 macOS 上,CUDA 库仅与 CUDA SDK 一起安装,并且可能不在库路径中。

返回:版本字符串(例如,“9.2”),如果未找到 CUDA,则返回 None。

cached_cuda_version()#

CUDA 检测系统的缓存版本。

conda_virtual_packages()#
_cuda_driver_version_detector_target(queue)#

尝试在子进程中检测操作系统中存在的 CUDA 版本。

在 Windows 和 Linux 上,CUDA 库由 NVIDIA 驱动程序包安装,并且通常在标准库路径中找到,而不是在 CUDA SDK 中(CUDA SDK 对于运行 CUDA 应用程序是可选的)。

在 macOS 上,CUDA 库仅与 CUDA SDK 一起安装,并且可能不在库路径中。

返回:版本字符串(例如,“9.2”),如果未找到 CUDA,则返回 None。

结果放在队列中,而不是返回值中。