sagemake studio lab, pytorch로 tensorboard 열기

sagemaker studio lab에서 tensorboard를 실행하는 방법을 기록해둔다.

 

 

1. tensorboardX, tensorboard 설치

pytorch에서 tensorboard를 사용하기 위해서는 tensorboardX를 설치해야 한다.

pip install tensorboardx

 

그리고 실제로 shell에서 tensorboard를 불러오기 위해서는 당연히 아래와 같이 tensorboard를 설치해야 한다.

pip install tensorboard

 

2. python 코드에서 tensorboard 추적대상 처리 & tensorboard 실제 실행

python 코드에서 tensorboard로 추적할 부분에 코드를 삽입하고, terminal에서 tensorboard log가 적재되는 경로로 들어가 아래와 같이 명령어를 입력해준다.

tensorboard --logdir .

 

728x90

 

3. URL 입력하여 tensorboard 확인

기존에 열려있는 lab의 주소를 복사해서 새 창에 붙여넣으면서 아래 색깔로 표시된 부분을 수정해준다.

{기존에 열려있는 lab 주소}

https://영어숫자로표현된부분.studio.us-east-2.sagemaker.aws/studiolab/default/jupyter/lab/tree/sample.ipynb

{tensorboard를 열기 위해 접근해야 하는 주소}

https://영어숫자로표현된부분.studio.us-east-2.sagemaker.aws/studiolab/default/jupyter/proxy/6006/

 

references

https://docs.aws.amazon.com/sagemaker/latest/dg/studio-tensorboard.html

Use TensorBoard in Amazon SageMaker Studio - Amazon SageMaker