맥 애플실리콘(m1~) 에서 pytorch 사용할 때 체크리스트

1. 아나콘다를 mac 에 호환이 되는 버전으로 설치할 것

2. pytorch 를 애플실리콘 호환되는 버전으로 설치할 것 (2.0 부터는 호환이 되는 것 같음)

3. 패키지들을 pip install 할 때 기존의 requirements.txt 중 arm64 버전에서 호환이 안되는 버전으로 적혀있으면 수정할 것.

4. cuda가 아닌 mps를 사용하기 위해 device 를 설정하는 변수들을 변경할 것.

 

4-1. device 설정할 때 다음과 같이 하면 된다.

device = torch.device('mps' if torch.backends.mps.is_available() else 'cpu')
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')