Update LD_LIBRARY_PATH instructions.

TuragaLab · Dec 11, 2024 · 57cc3e5 · 57cc3e5
1 parent 0569768
commit 57cc3e5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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.