맥북에서 M1 GPU 사용하기 위해 MPS backend 적용시 Conv3D 를 지원하지 않는 에러

혹시 예전에 mps backend 적용시에 conv3D 가 아직 지원안되어서 사용못한다는 글을 봤는데, 아직도 몇개는 지원이 안되는 건가요?
스크린샷 2023-02-11 오후 11.25.30

이렇게 에러가 계속 나네요 ㅠ 만약에 안되는 거면 사실상 파이토치를 mps를 사용해서 쓰기에는 힘들지 않을까 싶은데요..

1개의 좋아요

네, cuda와는 다르게 MPSROCm은 아직 시간이 더 필요해보이네요. :crying_cat_face:

그래도 ROCm보다는 빠르게 발전하고 있는 것 같아서 기대하고 있습니다.
(아직은 주로 서버에서 모델을 돌리고 있지만요ㅠ)

MPS 장치에서의 torch.nn.Conv3D()에 대한 이슈는 아래에서 관리 중인 것 같습니다.

아직 진행 중인 이슈 같은데요, 조금 더 기다려보셔야 할 것 같습니다 :sob: