Pytorch with AMD GPU on Linux

AMD에서 자사 CPU+GPU를 활용할수있는 ROCm 플렛폼을 제공함.

https://www.amd.com/ko/graphics/servers-solutions-rocm 

https://community.amd.com/t5/rocm/bg-p/amd-rocm-blog 

https://rocmdocs.amd.com/en/latest/ 

https://github.com/RadeonOpenCompute 

 

AMD  CPU+GPU용 파이토치 프론트엔드 + ROCm 백엔드로 제공됨

 

ROCm은 리눅스만 지원하며 바로 사용이 가능한 도커 컨테이너를 제공함

https://www.amd.com/en/technologies/infinity-hub/pytorch 

 

하지만 버전업데이트가 Pytorch 1.10 에서 멈췃음

 

Pytorch 에서도 공식적으로 ROCm 지원버전의 Pytorch 1.1를 제공함.

https://pytorch.org/get-started/locally/ 

 

Pytorch 에서 공식적으로 ROCm 지원버전의 Pytorch 1.8 지원함.  다만 직접 빌드해야함.

https://pytorch.org/blog/pytorch-1.8-released/#beta-amd-gpu-binaries-now-available  

https://github.com/pytorch/pytorch#from-source