1 minute read

이 메뉴얼은 수리계산실 컴퓨터의 Anaconda, CUDA, cuDNN을 활용한 Pytorch 환경을 구성하기 위한 메뉴얼입니다.

큰 틀은 Nvidia-driver 확인 및 설치, Anaconda 환경 구축, Pytorch 설치 순입니다. 이 메뉴얼도 같은 순서대로 구성되어있습니다.

0. PC

서버 컴퓨터 2대 1 를 제외하고 모든 컴퓨터의 비밀번호는
성균관입니다.

0-1. Ubuntu

2023년 7월 20일 기준 모든 Ubuntu 컴퓨터는 22.04.02 LTS

만약 사용하는 컴퓨터가 Windows 11라면 바로 Anaconda 항목으로 이동.

0-2. Terminal 사용법

Ubuntu 기반의 OS는 기본적으로 Terminal을 사용합니다.

Terminal 사용법

  • Ctrl + Alt + T을 누르면 Terminal 창이 뜹니다.
  • Terminal에서는 복사, 붙여넣기가 Ctrl + Shift + C, V입니다.
  • Terminal에서 Ctrl + C는 쓰고 있는 command를 취소합니다.

1. Nvidia Driver

1-1. 그래픽카드 확인하는 방법

nvidia-smi

nvidia-smi가 되면 정상적으로 작동하는 것입니다. Anaconda 항목으로 이동합니다.

정상적으로 작동하지 않는다면 Driver 설치로 이동합니다.

1-2. Driver 설치

nvidia-smi 입력했을 때 error 처럼 나오면 Driver를 설치2합니다.

sudo apt-get update
sudo ubuntu-drivers devices

입력하면 아래와 같은 결과값이 나옵니다. driver

이때 Recommended가 쓰여있는 nvidia-driver-{버전}을 확인하고

sudo apt-get install nvidia-driver-VERSION

이후 sudo reboot 하면 Driver 설치가 완료됩니다.


2. Anaconda

이후 편의를 위해 Anaconda는 Conda로 표기하겠습니다.

Terminal을 켜고 사진의 (base)가 보이면 Conda가 설치 되어있는 것입니다.

(base)가 보이지 않는다면 conda env list를 입력합니다.

  1. conda : command not found 라는 문구가 보이면 설치로 이동 conda not found
  2. 아니라면 conda env

2-1. 가상환경

2-1-1. 가상환경 생성

conda create -n 원하는 환경이름 

추가적인 parameter에 관해서는 docs참고
보통 원하는 python version을 parameter로 사용함

가상환경을 생성하고 나면

env list test 처럼 확인할 수 있습니다.

2-1-2. 가상환경 활성화

conda activate 환경이름

활성화를 하게되면 사진처럼 (base)가 (환경이름)으로 변경됩니다. 활성화 하지 않는다면 내가 작업하는 환경이 되지 않습니다. active

활성화 이후 PyTorch로 이동합니다.

2-2. Conda설치

sudo apt update
sudo apt install curl -y
curl --output anaconda.sh https://repo.anaconda.com/archive/Anaconda3-2022.10-Linux-x86_64.sh
sha256sum anaconda.sh
bash anaconda.sh

위의 코드를 차례대로 입력한뒤 입력하는 줄이 나오면 차례대로 Enter키/yes/Enter키/yes를 입력합니다.

source ~/.bashrc

입력하면 source 처럼 (base)가 등장합니다.


3. PyTorch 설치

conda environment가 활성화된 상태로 합니다.

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

에서 Linux, Conda, Python, CUDA 11.8을 선택하고 밑에 나온 Run this command를 복사해서 Terminal에 입력합니다. torch_official

대부분 이 과정에서 CUDA와 cuDNN가 설치됩니다.

확인

이 메뉴얼에서는 Jupyter lab을 이용해서 확인합니다.

conda install ipykernel
python -m ipykernel install --user --name=가상환경 이름
conda install jupyter -y
jupyter lab

을 차례대로 입력하면 jupyter창이 뜨고

현재 “test”라는 이름으로 만들었기 때문에 가장 위의 test를 눌러 아래의 코드들을 실행해보면 확인할 수 있습니다.

import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.backends.cudnn.enabled)
print(torch.backends.cuda.is_built())

pycuda 처럼 출력되면 pytorch-CUDA setting이 완료된것입니다.3(GPU는 컴퓨터마다 상이할 수 있습니다).


  1. 서버 컴퓨터 사용 문의는 fkrkty93@g.skku.edu 

  2. 대부분 여기서 끝남 

  3. GPU는 컴퓨터마다 상이할 수 있습니다.