새소식

Dev record

아나콘다 가상환경에 pytorch 간단 설치 import torch 에러

  • -
반응형

문제 상황

해당 환경에 pytorch 가 제대로 설치되지 않으면 위와 같이 pytorch가 import 되지 않는다.

여러 방법이 있지만 torch 버전에 상관없이 사용하고 있는 conda 환경에서 간단하게 쓰려면 아래 명령어로 쉽게 pytorch 설치가 가능하다.

가상환경에 pytorch 설치하기

conda install pytorch torchvision torchaudio cpuonly -c pytorch

 

명령어를 입력하면

Proceed ([y] / n) ? 라는 질문이 나오면 y를 입력한 후 엔터를 입력하면 설치가 진행된다.

 

이제 pytorch가 import 되는 것을 확인할 수 있다.

 

설치된 pytorch 버전 확인하기

간단하게 다음과 같은 코드로 확인 가능하다.

torch.__version__

현재 환경에는 pytorch가 2.1.0 버전이 설치되었다.

반응형
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.