Pytorch 설치

Mickey·2021년 11월 22일

Pytorch

목록 보기
1/1

Pytorch를 배워보고자 한다.

https://pytorch.kr/
위 사이트에서 배워보려고 한다.

2021년 11월 22일 기준으로 Python은 3.10.0이 Release 되었지만,
Pytorch를 사용하기위해서는 3.9.9를 설치해야 한다.
https://www.python.org/downloads/release/python-399/

CUDA 지원 버전을 사용할 것이므로 CUDA Toolkit을 설치해야 하는데,
오늘 기준으로 11.5.0 버전이 최신이지만, Pytorch에서는 11.3까지 지원하고 있어서 11.3.1을 설치한다.
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/cuda-11-3-1-download-archive

pip로 Pytorch를 설치한다.

https://pytorch.org/
Pytorch 공식 사이트 중단에


위 그림과 같이 설정하면

Run this Command 부분에 pip 설치 커맨드가 생성되는데, 이 내용을 복사하여 pip로 설치한다.

아래 사이트에서 whl 파일을 다운 받을 수 있다.
https://download.pytorch.org/whl/torch_stable.html
위 사이트에서 아래 파일을 다운로드 하여 설치한다.
cu113/torch-1.10.0%2Bcu113-cp39-cp39-win_amd64.whl

pip를 이용해 numpy를 설치하여야 pytorch를 사용 할 수 있다.

잘 설치 되었는지를 확인하기 위해 아래 링크를 참조 후 마지막 부분 내용을 실행해본다.
https://pytorch.org/get-started/locally/

Pytorch를 사용할 준비가 되었다.

profile
Mickey

0개의 댓글