config.py

TORCH_CUDNN_CONFIG = {
    "12": "9",
    "11": "8",
    "10": "7",
    "9": "7"
}

# cuda大版本（12.4）：cuda详细版本（12.4.1），cudnn版本（有可能为空）
NVIDIA_CUDA_CUDNN_VERSION_CONFIG = {
    "12.4": ["12.4.1", ""],
    "12.1": ["12.1.0", "8"]

}