DataRailroad

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