code정리(python)
[ PyTorch ] PyTorch 설치하기( with 아나콘다)
그린켐
2020. 12. 1. 11:25
Pytorch 설치 방법
1. PyTorch 홈페이지에서 pytorch 설치 commend 확인
pytorch.org/get-started/locally/
PyTorch
An open source deep learning platform that provides a seamless path from research prototyping to production deployment.
pytorch.org
Get Started를 클릭하면 화면에 START LOCALLY창이 보이고, 설정에 맞게 클릭 후 제일 하단에 있는 RUN this Command 를 확인한다.
※ CUDA 버전 확인 방법
| >>>nvcc -V |
or
| >>>nvidia-smi |
홈페이지에서 얻은 명령어를 가지고 pytorch를 설치한다. 2. 설치 진행
예시)
| >>> conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch |
다음 명령어로 정상적으로 설치 완료된 것을 확인해 볼 수 있다.
| >>> import pytorch |