Example #1
0
 def check(self, options, docker_registries_list):
     params = _get_params(self._config, options)
     client = self._get_client(params=params)
     try:
         details = common.check_environment(
             client, params['gpus'], params['log_dir'],
             docker_registries_list, self._config.get('requirements'),
             params.get('with_nvidia_docker'), False)
     finally:
         client.close()
     return details
Example #2
0
 def check(self, options):
     params = _get_params(self._config, options)
     client = common.ssh_connect_with_retry(params['server'],
                                            params['login'],
                                            self._config['privateKey'],
                                            login_cmd=params['login_cmd'])
     try:
         details = common.check_environment(
             client, params['gpus'], params['log_dir'],
             self._config['docker']['registries'],
             self._config.get('requirements'))
     finally:
         client.close()
     return details
Example #3
0
 def check(self, options):
     params = _get_params(self._config, options)
     client = common.ssh_connect_with_retry(
         params['host'],
         params['port'],
         params['login'],
         pkey=self._config.get('pkey'),
         key_filename=self._config.get('key_filename')
         or self._config.get('privateKey'),
         login_cmd=params['login_cmd'])
     try:
         details = common.check_environment(
             client, params['gpus'], params['log_dir'],
             self._config['docker']['registries'],
             self._config.get('requirements'), False)
     finally:
         client.close()
     return details