Error when running run_ray_train.py
Closed this issue · 1 comments
DemoreXTess commented
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 commented
Solved by adding site-packages/nvidia/cuda_runtime/lib
to LD_LIBRARY_PATH