4월쯤 베타 한다던 파이토치에서 메탈 백엔드를 지원하는 버전을 나이틀리로 공개 했습니다.

아직 conda 채널쪽엔 오류가 있는거같고 pip 설치로 1.12 버전 받으니 잘 작동합니다.

기존 cuda로 했던 부분에 device만 mps로 변경하면되서 코드 크게 고칠부분은 없어 보입니다.

.cuda() 처럼 .mps()가 아직은 따로 구현되진 않은듯 합니다.

https://pytorch.org/docs/master/notes/mps.html

모델 큰거 두개 올렸는데 아직 무난하긴 하네요

대신 에러도 낭낭합니다.

개인적으로 tf보다 토치를 더 쓰는데 안정화만 된다면 유용하게 써먹을거같네요