Error when running run_ray_train.py

[DemoreXTess]

File "/home/tess/anaconda3/envs/marlEnv/lib/python3.10/site-packages/ray/rllib/algorithms/algorithm_config.py", line 3623, in _check_if_correct_nn_framework_installed
raise ImportError(
ImportError: PyTorch was specified as the framework to use (via config.framework('torch'))! However, no installation was found. You can install PyTorch via pip install torch.

Ubuntu 20.04
Python 3.10.13 (from anaconda3)
Pytorch 2.0.1
Ray 2.6.1

[DemoreXTess]

Solved by adding site-packages/nvidia/cuda_runtime/lib to LD_LIBRARY_PATH