diff --git a/trains_agent/commands/worker.py b/trains_agent/commands/worker.py index bae0483..9fc9d84 100644 --- a/trains_agent/commands/worker.py +++ b/trains_agent/commands/worker.py @@ -1870,6 +1870,7 @@ class Worker(ServiceCommandSection): docker = 'docker' base_cmd = [docker, 'run', '-t'] + update_scheme = "" dockers_nvidia_visible_devices = 'all' gpu_devices = os.environ.get('NVIDIA_VISIBLE_DEVICES', None) if gpu_devices is None or gpu_devices.lower().strip() == 'all': diff --git a/trains_agent/session.py b/trains_agent/session.py index 618a71f..6473c49 100644 --- a/trains_agent/session.py +++ b/trains_agent/session.py @@ -177,7 +177,7 @@ class Session(_Session): folder_keys = ('agent.venvs_dir', 'agent.vcs_cache.path', 'agent.pip_download_cache.path', 'agent.docker_pip_cache', 'agent.docker_apt_cache') - singleton_folders = ('agent.venvs_dir', 'agent.vcs_cache.path',) + singleton_folders = ('agent.venvs_dir', 'agent.vcs_cache.path', 'agent.docker_apt_cache') if os.environ.get(ENV_TASK_EXECUTE_AS_USER): folder_keys = tuple(list(folder_keys) + ['sdk.storage.cache.default_base_dir'])