[STT][Whisper설치과정] #1 Python, NVIDIA Driver, PyTorch 설치하기

1. 시작

Whisper를 사용하기 위해 우선 그래픽 드라이버 및 PyTorch를 설치한다.

1.1. 설치 버전

Whisper를 사용하기 위해 아래와 같은 버전을 사용합니다.

아래와 같은 버전이 아닐 경우 2번으로 넘어가 설치합니다.

1.1.1. Python 버전

3.9.0 (최신버전으로 사용해도 무관)

• 아래 명령어로 Python 버전을 확인합니다.

python --version

1.1.2. CUDA 버전

11.8

• 아래 명령어로 CUDA 버전을 확인합니다.

nvcc --version

1.1.3. PyTorch

CUDA 11.8

• 아래 명령어로 CUDA 버전을 확인합니다.

python
# python 콘솔 진입
>>> import torch
>>> print(torch.__version__)
>>> exit() #python 콘솔 나가기

2. Python 설치

• 윈도우 기준 설치법입니다.
• 아래 링크로 들어가 최신버전을 설치해 줍니다.
• 저는 3.9.0 버전으로 실행했었지만 다른 컴퓨터에서 최신버전으로 실행해도 상관없이 실행했었습니다.

https://www.python.org/downloads/windows/

Python Releases for Windows

• 설치 실행파일을 실행하고 `Add python.exe to PATH`를 체크하고 Install Now를 클릭합니다.(환경변수를 등록합니다.)
• 터미널에서 `python`을 입력해봅니다. 그럼 아래와 같이 화면이 나오면 정상설치입니다.

$ python
Python 3.9.0 (tags/v3.9.0:9cf6752, Oct  5 2020, 15:34:40) [MSC v.1927 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>

python 설치전에 열어둔 터미널에다가 python을 설치 후 `python` 명령어를 입력하면 없다고 나옵니다.
새로운 터미널을 열어서 실행하세요.

3. CUDA 설치

• 윈도우 기준 설치법입니다.
• 아래 링크로 들어가 11.8 버전을 다운로드합니다.

https://developer.nvidia.com/cuda-11-8-0-download-archive

CUDA Toolkit 11.8 Downloads

• 각 버전에 맞게 선택 후 실행파일을 받습니다.
• Install type가 local인 경우는 설치 시 필요한 라이브러리까지 한 번에 받습니다.
• Install type가 network인 경우는 설치 실행 파일만 다운로드하고 라이브러리는 설치 실행파일을 실행하면 다운로드합니다.
• 설치 완료 후 터미널에서 `nvcc --version`을 입력합니다. 아래와 같이 나오면 성공적으로 설치한 겁니다.

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

4. PyTorch설치

• 윈도우 기준 설치법입니다.
• 아래 링크에 들어가 CUDA 11.8에 맞게 설정 후 커맨드를 입력합니다.(Package = pip)

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

PyTorch

• 설정을 하면 `Run this Command:`라고 pip에서 install 할 수 있는 명령어가 출력되는데, 해당 명령어를 복사하여 터미널에 붙여 넣은 다음 실행합니다.

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# CUDA 11.8 기준 링크

• 설치 완료 후 Python콘솔로 이동하여 버전을 확인합니다.

$ python
Python 3.9.0 (tags/v3.9.0:9cf6752, Oct  5 2020, 15:34:40) [MSC v.1927 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
2.0.1+cu118
>>> exit()