Pytorch를 설치하는 가장 간편한 방법은 conda를 이용하는 것이다. conda의 설치 및 사용 방법은 다음을 참조하길 바란다.
conda가 이미 있다면 아래의 명령을 실행하면 최신 버전의 pytorch가 설치된다. 이때 -c 옵션을 주는 것이 중요하다. 이는 pytorch채널에서 해당 라이브러리를 탐색하여 설치한다는 의미이다. 다른 pytorch 버전을 설치하고자하면, 공식 홈페이지를 참조하자.
conda install pytorch -c pytorch
TF의 tensorboard를 pytorch에서도 그대로 쓸 수 있다. SummaryWriter에 필요한 정보를 add하여 쉽게 그래프를 그릴 수 있다.
https://pytorch.org/docs/stable/tensorboard.html https://pytorch.org/tutorials/recipes/recipes/tensorboard_with_pytorch.html
tensorboard --logdir /tmp/tensorboard/exp1 --port 9000 --bind_all
또한 logdir을 절대경로로 설정해 주는 것이 여러모로 용이하다.
그리고 tensorboard에서는 logdir 아래 생성된 각 directory를 하나의 실험으로 간주해서 보여준다. 따라서 이 경로 아래에 개별 실험을 폴더 단위로 저장하면 쉽게 실험 별 성능 비교를 할 수 있다.
원래는 pip install tensorboard만 하면 잘 설치가 되어야한다. 그런데 nvidia-tensorboard 나 tensorboardX등이 뭔가 같이 설치되어 있는 경우, tensorboard가 정상 실행되지 않는 문제가 있다. 이러한 경우 일단 tensorboard와 관련된 설치를 모두 pip uninstall하고, 아래의 명령을 실행해서 깔끔하게 재설치한다.
conda install -c conda-forge tensorboard-plugin-wit
conda install -c conda-forge tensorboard
https://github.com/pytorch/pytorch/issues/22676#issuecomment-515662805