개발 꿀팁/PYTHON

pycharm 설치 pytorch

Jammie 2022. 11. 30. 11:59
반응형

pycharm에 pytorch를 설치할 때 많은 구덩이를 밟아 마침내 설치했습니다.다음은 튜토리얼입니다.

1.설치 전 준비 작업
콘다나 pip의 미러 소스를 국내 미러로 교체하는 것이 좋으며, 그렇지 않으면 엉뚱한 타임아웃 오류나 실패 등이 보고될 수 있다.
또한 python 버전을 3.6.2 이상으로 업그레이드하지 않으면 python 버전이 너무 낮으면 오류가 발생하므로 conda를 사용하여 가상 환경을 만들고 원하는 python 버전을 선택할 수 있습니다.
2. 쿠다 버전 확인
cuda 버전 보기 cmd 명령줄

nvidia-smi

3.명령줄로 설치
파이토치 홈페이지로 가기: Start Locally | 파이토치

스테이블 안정 버전을 선택하면 되고, os는 윈도우, 언어는 파이썬, 쿠다는 너와 같은 버전을 선택해.

물론 CPU 버전도 선택할 수 있지만 일반적으로 GPU를 달릴 수 있습니다.

pip 또는 conda를 사용하여 설치할 수 있습니다, 이것은 취향에 따라 다릅니다.

해당 pip 또는 conda가 설치한 명령을 pycharm 명령줄에 붙여 실행시키면 됩니다.

그 후 10분 또는 30분 정도 기다리면 설치가 완료됩니다

 

반응형