[PyTorch] AssertionError: Torch not compiled with CUDA enabled

소제목 입력

 

torch와 cuda 버전이 맞지 않아 출력되는 에러이다. 따라서, 공식 홈페이지에서 가이드대로 설치를 진행한다. 현재 Windows의 PyTorch는 Python 3.8 ~ 3.11만 지원하고 Python 2.x는 지원되지 않으므로 torch 가상 환경으로 파이썬 버전을 설치한다.

 

conda create --name torch python=3.9

 

torch 가상 환경을 활성화하고 CUDA 11.8 또는 12.1을 설치한다.

 

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

그 다음, GPU를 사용하는 확인하기 위해 device를 출력해 본다.

 

import torch

device = (
    "cuda"
    if torch.cuda.is_available()
    else "mps"
    if torch.backends.mps.is_available()
    else "cpu"
)

print(f"Using {device} device")

 

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

Start Locally