From 4ba6b3ed97d5cbe721f3ddd5f151e3699a97ca92 Mon Sep 17 00:00:00 2001 From: Faith Xu Date: Wed, 18 Aug 2021 16:54:08 -0700 Subject: [PATCH] Update CUDA-ExecutionProvider.md (#8778) --- docs/reference/execution-providers/CUDA-ExecutionProvider.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/reference/execution-providers/CUDA-ExecutionProvider.md b/docs/reference/execution-providers/CUDA-ExecutionProvider.md index 534dc235ef..d1e75b805c 100644 --- a/docs/reference/execution-providers/CUDA-ExecutionProvider.md +++ b/docs/reference/execution-providers/CUDA-ExecutionProvider.md @@ -22,13 +22,13 @@ Pre-built binaries of ONNX Runtime with CUDA EP are published for most language ## Requirements -Please reference table below for official GPU packages dependencies. +Please reference table below for official GPU packages dependencies for the ONNX Runtime inferencing package. Note that ONNX Runtime Training is aligned with PyTorch CUDA versions; refer to the Training tab on https://onnxruntime.ai/ for supported versions. |ONNX Runtime|CUDA|cuDNN|Notes| |---|---|---|---| |1.8|11.0.3|8.0.4 (Linux)
8.0.2.39 (Windows)|libcudart 11.0.221
libcufft 10.2.1.245
libcurand 10.2.1.245
libcublasLt 11.2.0.252
libcublas 11.2.0.252
libcudnn 8.0.4
libcupti.so 2020.1.1| |1.7|11.0.3|8.0.4 (Linux)
8.0.2.39 (Windows)|libcudart 11.0.221
libcufft 10.2.1.245
libcurand 10.2.1.245
libcublasLt 11.2.0.252
libcublas 11.2.0.252
libcudnn 8.0.4| -|1.5-1.6|10.2|8.0.3|CUDsA 11 can be built from source| +|1.5-1.6|10.2|8.0.3|CUDA 11 can be built from source| |1.2-1.4|10.1|7.6.5|Requires cublas10-10.2.1.243; cublas 10.1.x will not work| |1.0-1.1|10.0|7.6.4|CUDA versions from 9.1 up to 10.1, and cuDNN versions from 7.1 up to 7.4 should also work with Visual Studio 2017|