미니 파이토치 메탈 mps 백엔드가 공개됬습니다.
- idsr
- 조회 수 857
- 2022.05.19. 16:48
4월쯤 베타 한다던 파이토치에서 메탈 백엔드를 지원하는 버전을 나이틀리로 공개 했습니다.
아직 conda 채널쪽엔 오류가 있는거같고 pip 설치로 1.12 버전 받으니 잘 작동합니다.
기존 cuda로 했던 부분에 device만 mps로 변경하면되서 코드 크게 고칠부분은 없어 보입니다.
.cuda() 처럼 .mps()가 아직은 따로 구현되진 않은듯 합니다.
https://pytorch.org/docs/master/notes/mps.html
모델 큰거 두개 올렸는데 아직 무난하긴 하네요
대신 에러도 낭낭합니다.
개인적으로 tf보다 토치를 더 쓰는데 안정화만 된다면 유용하게 써먹을거같네요
댓글
0