PyTorch
새로운 PyTorch Profiler - 향상된 PyTorch 디버깅 프로파일러

파이토치(PyTorch)는 딥러닝 프로젝트를 빌드(build)하는 데 도움을 주는 파이썬 프로그램용 라이브러리 입니다.

기존 PyTorch 사용자는 autograd라는 프로파일러를 이용하여 PyTorch 작업 정보를 수집하였지만

포괄적인 GPU 정보를 수집하지 못하고 시각화할 수 있는 화면을 제공하지 않았습니다.

 

새로운 PyTorch Profiler는 GPU 및 PyTorch 작업 정보를 수집하고

모델의 병목현상을 감지하며 autograd 프로파일러와 호환성을 유지합니다.

아래 명령으로 설치하여 사용할 수 있습니다.

pip install torch_tb_profiler

 

또한 Visual Studio Code 와 통합 사용 가능합니다.

Visual Studio Code에서 "Launch TensorBoard"라는 패키지를 찾아서 설치하면 됩니다.

댓글 0