파이토치(PyTorch)는 딥러닝 프로젝트를 빌드(build)하는 데 도움을 주는 파이썬 프로그램용 라이브러리 입니다.
기존 PyTorch 사용자는 autograd라는 프로파일러를 이용하여 PyTorch 작업 정보를 수집하였지만
포괄적인 GPU 정보를 수집하지 못하고 시각화할 수 있는 화면을 제공하지 않았습니다.
새로운 PyTorch Profiler는 GPU 및 PyTorch 작업 정보를 수집하고
모델의 병목현상을 감지하며 autograd 프로파일러와 호환성을 유지합니다.
아래 명령으로 설치하여 사용할 수 있습니다.
pip install torch_tb_profiler
또한 Visual Studio Code 와 통합 사용 가능합니다.
Visual Studio Code에서 "Launch TensorBoard"라는 패키지를 찾아서 설치하면 됩니다.
pytorch
ai
profiler
댓글