Tüm Rehberler

PyTorch Kurulumu

Meta PyTorch derin öğrenme kütüphanesini kurun ve temel kullanımını öğrenin.

İleri30 dk

Kurulum Adımları

1. CPU sürümünü kurun:

pip install torch torchvision torchaudio

2. NVIDIA GPU için (CUDA 12.1):

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

3. Kurulumu doğrulayın:

python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

4. Basit model örneği:

python
import torch
import torch.nn as nn

class SimpleNet(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc1 = nn.Linear(784, 128)
        self.fc2 = nn.Linear(128, 10)
    
    def forward(self, x):
        x = torch.relu(self.fc1(x))
        return self.fc2(x)

model = SimpleNet()
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=0.001)

5. GPU'ya taşıma:

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = model.to(device)

6. Model kaydetme:

torch.save(model.state_dict(), 'model.pth')

7. Model yükleme:

model.load_state_dict(torch.load('model.pth'))