Pytorch의 Vulkan backend 사용

Pytorch가 1.7 버전부터 Vulkan 을 backend로 사용할 수 있는 기능을 제공하고 있었네요. 1.7버전이면 꽤 오래된거 같은데 아직 prototype feature 인 걸 보면 기존의 Cuda ecosystem을 따라잡긴 역시나 쉽지 않은가 봅니다. AMD계열 GPU나 Android 등에서도 Vulkan 사용이 가능한걸로 아는데, 그러면 앞으로 Nvidia GPU아닌 다른 GPU를 좀 더 잘 활용할 수 있게 될까요? 아직은 많은 모델들이 Cuda 기반으로 작성이 되어서 어렵겠지만, 앞으로 좀 더 다양한 하드웨어에서 딥러닝 모델들이 GPU 가속을 사용할 수 있게 되는걸까요? 실제로 Pytorch를 Vulkan backend로 AMD GPU등에서 사용해 본 사례가 궁금하네요..ㅎㅎ

PyTorch Vulkan Backend User Workflow — PyTorch Tutorials 1.13.1+cu117 documentation

PyTorch

PyTorch Vulkan Backend User Workflow — PyTorch Tutorials 1.13.1+cu117 documentation

더 많은 콘텐츠를 보고 싶다면?

또는

이미 회원이신가요?

2023년 1월 20일 오후 3:48

 • 

저장 1조회 2,323

댓글 0