Skip to main content

One post tagged with "M2"

View All Tags

PyTorch on m2 mac

· 5 min read
Ryukato
BackEnd Software Developer

Apple Silicon(M1, M2, M3 등)을 사용하는 macOS 환경에서는 PyTorch의 cuda 대신 Metal Performance Shaders(MPS) 백엔드를 사용해야 합니다. 하지만 이때 주의해야 할 점 중 하나가 바로 torch_dtype 설정입니다.