Apertura de TensorBoard con el módulo sagemaker.interactive_apps.tensorboard

El módulo sagemaker.interactive_apps.tensorboard ofrece una función llamada get_app_url que genera direcciones URL prefirmadas o sin firmar para abrir la aplicación TensorBoard en cualquier entorno de SageMaker AI o Amazon EC2. El objetivo es proporcionar una experiencia unificada tanto a los usuarios de Studio Classic como a los que no lo son. Para el entorno Studio, puede abrir TensorBoard ejecutando la función get_app_url() tal como está, o también puede especificar un nombre de trabajo para iniciar el seguimiento cuando se abra la aplicación TensorBoard. Para entornos que no sean de Studio Classic, para abrir TensorBoard, proporcione la información de su dominio y de su perfil de usuario a la función de utilidad. Con esta funcionalidad, independientemente de dónde o cómo ejecute el código de entrenamiento y lance los trabajos de entrenamiento, puede acceder directamente a TensorBoard ejecutando la función get_app_url en su terminal o cuaderno de Jupyter.

Opción 1: para SageMaker AI Studio Classic

Si utiliza SageMaker Studio Classic, puede abrir directamente la aplicación TensorBoard o recuperar una URL sin firmar ejecutando la función get_app_url de la siguiente manera: Como ya se encuentra en el entorno de Studio Classic y ha iniciado sesión como usuario del dominio, get_app_url() genera una URL sin firmar porque no es necesario volver a autenticarse.

Para abrir la aplicación TensorBoard

El siguiente código abre automáticamente la aplicación TensorBoard desde la URL sin firmar que la función get_app_url() devuelve en el navegador web predeterminado de su entorno.

from sagemaker.interactive_apps import tensorboard

region = "us-west-2"
app = tensorboard.TensorBoardApp(region)

app.get_app_url(
    training_job_name="your-training_job_name" # Optional. Specify the job name to track a specific training job 
)

Para recuperar una URL sin firmar y abrir la aplicación TensorBoard manualmente

El siguiente código imprime una URL sin firmar que puede copiar en un navegador web y abrir la aplicación TensorBoard.

from sagemaker.interactive_apps import tensorboard

region = "us-west-2"
app = tensorboard.TensorBoardApp(region)
print("Navigate to the following URL:")
print(
    app.get_app_url(
        training_job_name="your-training_job_name", # Optional. Specify the name of the job to track.
        open_in_default_web_browser=False           # Set to False to print the URL to terminal.
    )
)

Tenga en cuenta que si ejecuta los dos ejemplos de código anteriores fuera del entorno de SageMaker AI Studio Classic, la función devolverá una URL a la página de inicio de TensorBoard en la consola de SageMaker AI, ya que no contienen información de inicio de sesión en su dominio ni en su perfil de usuario. Para crear una URL prefirmada, consulte la opción 2 en la siguiente sección.

Opción 2: para entornos que no sean de Studio Classic

Si utiliza entornos que no son de Studio Classic, como la instancia de cuaderno de SageMaker o Amazon EC2, y desea abrir TensorBoard directamente desde el entorno en el que se encuentra, debe generar una URL prefirmada con la información de su dominio y perfil de usuario. Una URL prefirmada es una URL en la que se inicia sesión en Amazon SageMaker Studio Classic mientras se crea la URL con su dominio y perfil de usuario y, por lo tanto, se le concede acceso a todos los archivos y aplicaciones de dominio asociados a su dominio. Para abrir TensorBoard a través de una URL prefirmada, use la función get_app_url con el nombre de su dominio y perfil de usuario de la siguiente manera.

Tenga en cuenta que esta opción requiere que el usuario del dominio cuente con el permiso sagemaker:CreatePresignedDomainUrl. Sin el permiso, el usuario del dominio recibirá un error de excepción.

print(
    app.get_app_url(
        training_job_name="your-training_job_name", # Optional. Specify the name of the job to track.
        create_presigned_domain_url=True,           # Reguired to be set to True for creating a presigned URL.
        domain_id="your-domain-id",                 # Required if creating a presigned URL (create_presigned_domain_url=True).
        user_profile_name="your-user-profile-name", # Required if creating a presigned URL (create_presigned_domain_url=True).
        open_in_default_web_browser=False,          # Optional. Set to False to print the URL to terminal.
        optional_create_presigned_url_kwargs={}     # Optional. Add any additional args for Boto3 create_presigned_domain_url
    )
)

optional_create_presigned_url_kwargs={"ExpiresInSeconds": 1500}

Navigate to the following URL:
https://us-west-2.console.aws.amazon.com/sagemaker/home?region=us-west-2#/tensor-board-landing