From 57cc3e5a080061a8bf22f7fc7ac8a673eb44ef57 Mon Sep 17 00:00:00 2001
From: Roman Vaxenburg <46580141+vaxenburg@users.noreply.github.com>
Date: Tue, 10 Dec 2024 19:22:04 -0500
Subject: [PATCH] Update `LD_LIBRARY_PATH` instructions.

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 4d845c2..becfba0 100644
--- a/README.md
+++ b/README.md
@@ -122,7 +122,8 @@ Follow these steps to install `flybody`:
    ```
 2. Also, for the ML and Ray extensions, `LD_LIBRARY_PATH` may require an update, e.g.:
    ```bash
-   export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/your/path/to/miniconda3/envs/flybody/lib
+   CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
+   export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
    ```
 
 3. You may want to run `pytest` to test the main components of the `flybody` installation.