개인 공부/환경세팅
Tensorboard 사용법 익히기
_h.j
2021. 12. 17. 13:32
728x90
우분투 18.04, 파이참 환경
가상환경 생성
환경세팅
가상환경 생성

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