PyTorch

Instal·la UV:

curl -LsSf https://astral.sh/uv/install.sh | sh
source $HOME/.local/bin/env

uv init --python 3.12 torx
cd torx
uv add torch torchvision

PyTorch - Start locally

import torch

print("cuda", torch.cuda.is_available())

x = torch.rand(5, 3)
print(x)

import torch
from torchvision import datasets,transforms
from torch.utils.data import DataLoader

is_gpu = torch.cuda.is_available()

print('is_gpu', is_gpu)

transform = transforms.Compose([
    transforms.ToTensor()
])

train = datasets.MNIST('',train = True, download = True, transform=transform)
test = datasets.MNIST('',train = False, download = True, transform=transform)

trainloader = DataLoader(train, batch_size= 32, shuffle=True)
testloader = DataLoader(test, batch_size= 32, shuffle=True)

TODO

• Using uv with PyTorch
• Training and Testing a Basic Neural Network using Pytorch