Python

CUDA

check cuda version:

nvcc --version

check cudnn version:

python -c "import torch; print(torch.backends.cudnn.version())"

check cuda torch is installed:

python -c "import torch; print(torch.version.cuda)"

check onnxruntime is installed:

python -c "import onnxruntime; print(onnxruntime.__version__)"

or

pip show onnxruntime-gpu