在安装 onnxruntime-gpu 选择版本时需要查看本机 CUDA 和 cuDNN 版本。

查看 CUDA 和 cuDNN 版本

import platform
import torch

print("python.version:", platform.python_version())
print("torch.version:", torch.__version__)
print("CUDA.version:", torch.version.cuda)
print("cuDNN.version:", torch.backends.cudnn.version())

输出日志如下：

python.version: 3.10.14
torch.version: 2.4.0+cu121
CUDA.version: 12.1
cuDNN.version: 90100