pytorch로 gpu 사용하기

pip install torch==2.4.0+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

https://download.pytorch.org/whl/cu118

 

이렇게 cuda를 붙여서 설치한 다음

 

import torch

print(torch.cuda.is_available())

import torch

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

print(device)

이 코드로 확인해보자 

true라 뜨면 잘 되는것이다.