Introducing PyTorch Profiler – The New And Improved Performance Debugging Profiler For PyTorch

The analysis and refinement of the large-scale deep learning model's performance is a constant challenge that increases in importance with the model’s size. Owing to a lack of available resources, PyTorch users had a hard time overcoming this problem. There were common GPU hardware-level debugging tools, but PyTorch-specific background of operations was not available. Users had to merge multi-tools or apply minimal correlation information manually to make sense of the data to retrieve the missing information. The PyTorch Profiler came to the rescue, an open-source tool for precise, efficient, and troubleshooting performance investigations of large-scale deep learning models. What is

www.marktechpost.com

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

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

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

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

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

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

pip install torch_tb_profiler

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

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

#profiler

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