본문으로 건너뛰기

"MPS" 태그로 연결된 1개 게시물개의 게시물이 있습니다.

모든 태그 보기

PyTorch on m2 mac

· 약 5분
Ryukato
BackEnd Software Developer

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