CUDA가 아닌 MPS로 MNIST학습
저는 M1 Air를 사용하고 있습니다. 하지만 M1 Air는 apple사의 칩이라서 NVIDIA의 CUDA를 사용하지 못합니다. 하지만 pytorch 1.12부터는 CUDA대신 MPS를 활용해서 M1의 GPU를 사용할 수 있습니다. 1. pytorch를 다운로드 받습니다. 아래의 링크에서 자기 PC와 맞는 사양을 선택합니다. https://pytorch.kr/get-started/locally/ 파이토치 한국 사용자 모임 (PyTorch Korea User Group) 파이토치 한국 사용자 모임에 오신 것을 환영합니다. 딥러닝 프레임워크인 파이토치(PyTorch)를 사용하는 한국어 사용자들을 위해 문서를 번역하고 정보를 공유하고 있습니다. pytorch.kr 저는 M1이고 pip를 사용하므로 아래와 같습..
공부/AI
2022. 6. 23. 20:59
