m1 macbook에서 GPU 가속

givemcdonalds·2023년 7월 3일
0

Microsoft

목록 보기
9/9

MPS(Multi-Process Service)

mps가 torch에 활성화 되어있는지 확인하기

% python
Python 3.11.3 (main, Apr 19 2023, 18:49:55) [Clang 14.0.6 ] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
im>>> import math
>>> print(torch.backends.mps.is_available())
True
>>> print(torch.backends.mps.is_built())
True
>>> exit()

cuda 대신 mps 사용

profile
운동 좋아하는 개발자

0개의 댓글