From 989967e2b2829d43caf6d8b2dbea4dec333f74d1 Mon Sep 17 00:00:00 2001 From: davidpagnon Date: Fri, 6 Sep 2024 15:34:47 +0200 Subject: [PATCH] formatting --- README.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 49a100e..27b4f79 100644 --- a/README.md +++ b/README.md @@ -133,11 +133,12 @@ If you don't use Anaconda, type `python -V` in terminal to make sure python>=3.9 Run `nvidia-smi` in a terminal. If this results in an error, your GPU is probably not compatible with CUDA. If not, note the "CUDA version": it is the latest version your driver is compatible with (more information [on this post](https://stackoverflow.com/questions/60987997/why-torch-cuda-is-available-returns-false-even-after-installing-pytorch-with)). - Then go to the [ONNXruntime requirement page](https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements), note the latest compatible CUDA and cuDNN requirements. Finally, go to the [pyTorch website]( https://pytorch.org/get-started/locally) and install the latest version that satisfies these requirements (beware that torch 2.4 ships with cuDNN 9, while torch 2.3 installs cuDNN 8). You may need to opt for a previous version, downloadable [there](https://pytorch.org/get-started/previous-versions/). For example: + Then go to the [ONNXruntime requirement page](https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements), note the latest compatible CUDA and cuDNN requirements. Next, go to the [pyTorch website](https://pytorch.org/get-started/previous-versions/) and install the latest version that satisfies these requirements (beware that torch 2.4 ships with cuDNN 9, while torch 2.3 installs cuDNN 8). For example: ``` cmd - pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 + pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 ``` - Then install ONNX Runtime with GPU support: + + Finally, install ONNX Runtime with GPU support: ``` pip install onnxruntime-gpu ```