GPU server setting #4_pytorch 설치(Cuda v 11.2) 및 커널에 가상환경 추가하기.

Pytorch 설치

• https://sanglee325.github.io/environment/pytorch-cuda112/#solution 참조
• pytorch 사이트에 접속하면 11.0과 11.3만 존재합니다.
• 참조 블로그를 읽어보니 11.0 일 때 torch가 잘만 import 되더라 라는 말이 있어 v11.0으로 그냥 설치하였습니다.

 

쥬피터 노트북과 파이썬 커널에 내 가상환경 추가하기

• 내 프로젝트 폴더 내에서 가상환경에 들어갑니다.

source ./venv/bin/activate

• 쥬피터 노트북에 내 가상환경을 설정합니다.

python -m ipykernel install --user --name 가상환경이름 --display-name "커널 이름"

 

• VS code 에서는 kernel select 를 누르면 python:jupyternotebook 의 extension을 설치하겠는지를 묻고, 설치하면 위 코드에서 정한 커널이름이 선택지 중에 하나로 뜨게 됩니다.
• 뜨지 않는다면 다음을 참고하면 될 것 같습니다. (https://stackoverflow.com/questions/44109867/using-multiple-python-kernels-in-visual-studio-code)

 

 

 

Torch 설치 확인하기

import torch
torch.cuda.is_available()
torch.cuda.get_device_name()