문제 상황
해당 환경에 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 버전이 설치되었다.