nemo_rl.utils.prefetch_venvs

Module Contents

Functions

prefetch_venvs	Prefetch all virtual environments that will be used by workers.
create_frozen_environment_symlinks	Create python-{ClassName} wrapper scripts in /usr/local/bin for frozen environment support.

API

nemo_rl.utils.prefetch_venvs.prefetch_venvs()

Prefetch all virtual environments that will be used by workers.

nemo_rl.utils.prefetch_venvs.create_frozen_environment_symlinks(venv_configs)

Create python-{ClassName} wrapper scripts in /usr/local/bin for frozen environment support.

Only runs in container (when NRL_CONTAINER=1 is set).

Parameters:

venv_configs – Dictionary mapping py_executable to list of actor FQNs