반응형
module 'torch.backends' has no attribute 'mps' 해결 방법
* 현재 Apple M1 pro 사용하고 있습니다.
반응형
상황
- import torch 이후 torch.device('mps:0' if torch.backends.mps.is_available() else 'cpu')로 mps가 잘 되었다가 하루 뒤
module 'torch.backends' has no attribute 'mps'가 뜨면서 작동하지 않았습니다.
해결
저는 일단 Pytorch version 확인 후 update 할려고 했지만 실패해서 다른 방법을 찾다 해결했습니다.
1. 가상환경 또는 사용하는 환경에서
$ pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
진행해서 설치 했습니다.
해결한 상태에서 확인 한 결과입니다.
import platform
print(platform.platform())
import torch
print(torch.__version__)
-> 2.1.0.dev20230407
print(torch.backends.mps.is_built())
-> True
반응형
'머신러닝 & 딥러닝 > 설정' 카테고리의 다른 글
Apple M1 GPU 사용하는 방법 (0) | 2023.04.08 |
---|