From 384c686f4015e3e92a555156aedbc35a8adebf21 Mon Sep 17 00:00:00 2001 From: Pranav Sharma Date: Tue, 22 Oct 2019 07:21:48 -0700 Subject: [PATCH] Update README and other files with the correct cuda version used for 1.0 release. (#2222) --- BUILD.md | 2 +- README.md | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/BUILD.md b/BUILD.md index 52d6100b04..99a3308430 100644 --- a/BUILD.md +++ b/BUILD.md @@ -139,7 +139,7 @@ Build command: ### CUDA #### Pre-Requisites * Install [CUDA](https://developer.nvidia.com/cuda-toolkit) and [cuDNN](https://developer.nvidia.com/cudnn) - * ONNX Runtime is built and tested with CUDA 10.0 and cuDNN 7.3 using the Visual Studio 2017 14.11 toolset (i.e. Visual Studio 2017 v15.3). 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. + * ONNX Runtime is built and tested with CUDA 10.0 and cuDNN 7.6 using the Visual Studio 2017 14.11 toolset (i.e. Visual Studio 2017 v15.3). 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. * The path to the CUDA installation must be provided via the CUDA_PATH environment variable, or the `--cuda_home parameter` * The path to the cuDNN installation (include the `cuda` folder in the path) must be provided via the cuDNN_PATH environment variable, or `--cudnn_home parameter`. The cuDNN path should contain `bin`, `include` and `lib` directories. * The path to the cuDNN bin directory must be added to the PATH environment variable so that cudnn64_7.dll is found. diff --git a/README.md b/README.md index 27282b44d9..b42b1999e7 100644 --- a/README.md +++ b/README.md @@ -103,8 +103,7 @@ system. [vc_redist.x64.exe](https://aka.ms/vs/16/release/vc_redist.x64.exe) and [vc_redist.x86.exe](https://aka.ms/vs/16/release/vc_redist.x86.exe) * For Linux, the system must have **libgomp.so.1** which can be installed using `apt-get install libgomp1`. * GPU builds require CUDA runtime libraries being installed on the system: - * Version: **CUDA 10.0** and **cuDNN 7.3** - * Linux Python packages require **CUDA 10.1** and **cuDNN 7.6** + * Version: **CUDA 10.0** and **cuDNN 7.6** * Older ONNX Runtime releases: used **CUDA 9.1** and **cuDNN 7.1** - please refer to [prior release notes](https://github.com/microsoft/onnxruntime/releases) for more details. * Python binaries are compatible with **Python 3.5-3.7**. See [Python Dev Notes](./docs/Python_Dev_Notes.md). If using `pip` to be download the Python binaries, run `pip install --upgrade pip` prior to downloading. * Certain operators makes use of system locales. Installation of the **English language package** and configuring `en_US.UTF-8 locale` is required.