맥북 M1 Chip GPU에서 pytorch 실행
맥북 M1 Chip는 Nvidia 그래픽카드를 사용하지 않는다. Apple 자체 개발한 M1 Chip을 사용하며, 이 아키텍처는 CPU와 GPU가 메모리를 공유하는 구조로 되어 있다. 따라서, NVIDA 그래픽카드 메모리에서 구동하도록 만들어진 CUDA 를 사용하지 못한다. pytorch + GPU를 사용하는 대부분 실험과 Application이 CUDA를 사용하는 것을 고려할 때, 이것은 매우 심각한 문제라고 할 수 있다. 그러나, 애플에서 이부분을 고려하지 않을리 없다. 애플 또한 CUDA에 해당하는, M1 Chip의 GPU가속을 사용하도록 지원하는 프레임워크 "MPS"를 지원하고, pytorch 또한 이 MPS를 지원한다. MPS가 세팅되었는지 확인하기 위해서 다음과 `torch.backends.m..
개발 이야기/Python
2023. 11. 13. 17:33
