새소식

Study/시행착오

(해결) M1 Mac에서 GPU를 사용해 pytorch 사용하기

  • -
728x90

시도해보려는 프로젝트의 이미지 크기가 어마어마해서 GPU로 해보려고 이것저것 시도해보는 중..입니다만
 cuda 버전에 맞는 torch를 잘 설치해주어야 한다 에서 실패한 것 같다? 
가 아니라, 이 링크 를 보고 설치한지라 mps로 사용하려고 설치했던 것이 기억이 났습니다. 

설치할 때 무엇을 염두에 두고 진행했는지 항상 기억 혹은 기억해두어야겠습니다 ^^ 🔥

728x90
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.