parameters: (environ_cp)
docstring: Set TF_NCCL_VERSION.
docstring_summary: Set TF_NCCL_VERSION.
docstring_tokens: Set TF_NCCL_VERSION .
function: def set_tf_nccl_version(environ_cp):
    """Set TF_NCCL_VERSION."""
    if not is_linux():
      raise ValueError('Currently NCCL is only supported on Linux platform.')
  
    if 'TF_NCCL_VERSION' in environ_cp:
      return
  
    ask_nccl_version = (
        'Please specify the locally installed NCCL version you want to use. '
        '[Leave empty to use http://github.com/nvidia/nccl]: ')
    tf_nccl_version = get_from_env_or_user_or_default(environ_cp,
                                                      'TF_NCCL_VERSION',
                                                      ask_nccl_version, '')
    environ_cp['TF_NCCL_VERSION'] = tf_nccl_version
function_tokens: def set_tf_nccl_version ( environ_cp ) : if not is_linux ( ) : raise ValueError ( 'Currently NCCL is only supported on Linux platform.' ) if 'TF_NCCL_VERSION' in environ_cp : return ask_nccl_version = ( 'Please specify the locally installed NCCL version you want to use. ' '[Leave empty to use http://github.com/nvidia/nccl]: ' ) tf_nccl_version = get_from_env_or_user_or_default ( environ_cp , 'TF_NCCL_VERSION' , ask_nccl_version , '' ) environ_cp [ 'TF_NCCL_VERSION' ] = tf_nccl_version