Python_Error

Pytorch 설치 시 주의 사항(ERROR: Failed building wheel for torch 에러 해결)

qlsenddl 2020. 6. 17. 14:13
728x90

cmd 창에

>> python -m pip install torch

명령으로 pytorch를 설치하려고 하면 아래와 같은 에러가 발생하는 문제가 있다.

ERROR: Failed building wheel for torch

 

해결 방법

pytorch의 경우 python -m pip install torch 명령으로 설치하는 것이 아니라,

https://pytorch.org

 

PyTorch

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.

pytorch.org

링크로 들어가서 명령문을 찾아야 한다.

해당 링크로 들어가면 위 화면이 나타나고 맞는 사양을 클릭하면 그 사양에 맞는 명령문이 Run this Command에 표시된다. 해당 코드를 복사해서 cmd에 붙여넣기를 하면 pytorch가 설치된다.

Pytorch Build의 경우 Preview(Nightly)는 일종의 베타버전이다. 그러므로 Stable(1.5)를 선택해서 사용하면 된다.

728x90