개인 공부/환경세팅

Tensorboard 사용법 익히기

_h.j 2021. 12. 17. 13:32
728x90

우분투 18.04, 파이참 환경

 

가상환경 생성

 

[pytorch로 tensorboard 따라하기]

 

환경세팅

가상환경 생성

pytorch 설치, tensorboard 설치

이 사이트로 가서 버전에 맞게 설치

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

pip install tensorboard

 

scalr 해보기

tensorboard 결과 확인

tensorboard --logdir=runs

 

 

 

Tensorboard로 모델,데이터,학습 시각화하기

공홈에 나온 것 처럼 

from torch.utils.tensorboard import SummaryWriter

# 기본 `log_dir` 은 "runs"이며, 여기서는 더 구체적으로 지정하였습니다
writer = SummaryWriter('runs/fashion_mnist_experiment_1')

이렇게 import하면 에러난다.

 

여기 사이트에서 처럼

tensorboardX 설치 후 tensorboardX를  import한다.

 

 

tensorboardX가서 add_mesh든 이미지든 찾아서 사용해보자

 

 

 

 

 

 

 

728x90