sagemaker.core.instance_types_gpu_info#
Accessors to retrieve instance types GPU info.
Functions
|
Retrieves instance types GPU info of the given region. |
- sagemaker.core.instance_types_gpu_info.retrieve(region: str) Dict[str, Dict[str, int]][source]#
Retrieves instance types GPU info of the given region.
- Parameters:
region (str) – The AWS region.
- Returns:
- A dictionary that contains instance types as keys
and GPU info as values or empty dictionary if the config for the given region is not found.
- Return type:
dict[str, dict[str, int]]
- Raises:
ValueError – If no config found.