Skip to main content

​
PyTorch CUDA 설치

PyTorch는 CUDA가 포함된 wheel을 설치하는 방식이 가장 단순하고 안정적입니다.

​
사전 준비

​
설치 과정

1

가상환경 준비

mkdir -p ~/dev/torch-cuda-test
cd ~/dev/torch-cuda-test
uv venv
source .venv/bin/activate
2

PyTorch CUDA 패키지 설치

uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
3

GPU 인식 확인

python - <<'PY'
import torch
print("torch:", torch.__version__)
print("cuda available:", torch.cuda.is_available())
print("device count:", torch.cuda.device_count())
if torch.cuda.is_available():
    print("device 0:", torch.cuda.get_device_name(0))
PY

​
설치 확인하기

cuda available: True가 출력되면 GPU 가속이 정상입니다.

​
문제 해결

  • cuda available: False: 드라이버 설치 상태(nvidia-smi)를 먼저 확인하세요.
  • 버전 호환 문제: PyTorch 인덱스 URL의 CUDA 버전(cu124)을 환경에 맞게 조정하세요.