PyTorch คืออะไร และวิธีติดตั้งบน Cloud Server

ทดสอบบน Ubuntu Server 24.04 LTS

1. 1

   ติดตั้ง Python และเครื่องมือพื้นฐาน

   คัดลอก

   sudo apt update
   sudo apt -y install python3-pip python3-venv build-essential
   python3 -m venv ~/venv && source ~/venv/bin/activate

2. 2

   ติดตั้งไดรเวอร์ GPU ถ้ามีการ์ด

   ตรวจว่าไดรเวอร์ทำงานถูกต้องก่อนติดตั้งไลบรารี เพราะเป็นสาเหตุของปัญหาที่พบบ่อยที่สุด

   คัดลอก

   sudo apt -y install nvidia-driver-550
   sudo reboot
   nvidia-smi

3. 3

   ติดตั้ง PyTorch

   เลือกคำสั่งติดตั้งให้ตรงกับรุ่นของไดรเวอร์จากเว็บทางการ เพราะเวอร์ชันที่ไม่ตรงกันจะทำให้ใช้ GPU ไม่ได้

   คัดลอก

   pip install torch torchvision torchaudio

4. 4

   ตรวจว่าใช้ GPU ได้จริง

   คัดลอก

   python3 -c "import torch; print(torch.__version__, torch.cuda.is_available(), torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU only')"

5. 5

   ทดสอบฝึกโมเดลอย่างง่าย

   คัดลอก

   import torch
   import torch.nn as nn
   
   model = nn.Sequential(nn.Linear(10, 64), nn.ReLU(), nn.Linear(64, 1)).cuda()
   x = torch.randn(256, 10).cuda()
   y = torch.randn(256, 1).cuda()
   opt = torch.optim.Adam(model.parameters(), lr=1e-3)
   
   for step in range(100):
       loss = ((model(x) - y) ** 2).mean()
       opt.zero_grad(); loss.backward(); opt.step()
   print("loss:", loss.item())

6. 6

   ตั้งการรันงานฝึกเป็นบริการ

   สำหรับงานฝึกที่ใช้เวลานาน ให้รันผ่านตัวจัดการเซสชันหรือเป็นบริการของระบบ เพื่อไม่ให้งานหยุดเมื่อตัดการเชื่อมต่อ SSH

เลือกรุ่นของไลบรารีให้ตรงกับไดรเวอร์ GPU เสมอ เพราะเวอร์ชันที่ไม่ตรงกันเป็นสาเหตุอันดับหนึ่งที่ทำให้ใช้ GPU ไม่ได้ทั้งที่ติดตั้งสำเร็จ