<aside> 💡

# 프로젝트 폴더로 이동

cd 원하는경로

# 가상환경 생성

python -m venv venv

# 가상환경 활성화 (Windows)

venv\Scripts\activate

# PyTorch GPU 버전 설치 (CUDA 12.1 예시)

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

</aside>

매번 activate만 한 다음, 파이썬 파일 내에 다음 내용 입력

import torch
import torch.nn as nn

# GPU 설정
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(f"사용 중인 디바이스: {device}")

# 모델 정의
model = nn.Linear(10, 1)
model = model.to(device)  # 모델을 GPU로

# 데이터
data = torch.randn(32, 10)
data = data.to(device)  # 데이터를 GPU로

# 학습 코드..