Conda 가상환경으로 PyTorch 설치하기(Window 버전)

Pytorch는 cpu, gpu 둘다 사용 가능하기에 우선적으로 컴퓨터의 nvidia의 cuda 버전을 확인해야한다.

 

!nvcc --version

 

문과 출신 신의 노트북 cuda 버젼

 

코드를 입력하고 나온 정보에서 release 다음에 적혀있는 버전이 cuda 버전이다. 해당 버전을 기억하고 다음 pytorch 사이트로 이동한다.

https://pytorch.org/get-started/locally/

PyTorch

사이트 내의 모습

사이트를 들어가서 개발환경에 맞는 옵션들을 설정해준다. 여기서 우린 Conda 가상환경에 만들것이기에 Package를 Conda로 설정해 준다. 이전에 봐두었던 CUDA 버전도 넣어주면, 아래에 설치 코드가 나타난다. 이 코드를 저장해둔다.

 

이제 가상환경 만들기를 시작한다.

 

우선, Anaconda Prompt를 연다.

 

conda create -y -n pytorch ipykernel

 

여기서 pytorch 대신 원하는 가상환경의 다른 이름을 넣어도된다.

 

 

설정해준 가상환경이름을 activate 코드 다음에 넣어 가상 환경 안으로 들어간다.

 

activate pytorch

 

이전에 기억해둔 pytorch 설치 코드를 입력해준다.

 

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

 

그 후 Jupyter Notebook에 새 kernel을 등록한다.

 

python -m ipykernel install --user --name pytorch --display-name "PyTorch"

 

여기서도 "PyTorch" 대신 Jupyter Notebook에 표시될 kernel의 다른 이름으로 설정 가능하다.

 

가상 환경 빠져 나오면 끝이다.

 

deactivate