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"라는 패키지를 찾아서 설치하면 됩니다.
아직 댓글이 없습니다. 첫번째 댓글 작성자가 되어주세요 :)