Google Colab - ใช้ GPU ฟรี

Google ให้การใช้ GPU ฟรีสำหรับโน้ตบุ๊ก Colab ของคุณ

การเปิดใช้งาน GPU

ในการเปิดใช้งาน GPU ในโน้ตบุ๊กของคุณให้เลือกตัวเลือกเมนูต่อไปนี้ -

Runtime / Change runtime type

คุณจะเห็นหน้าจอต่อไปนี้เป็นผลลัพธ์ -

เลือก GPUและโน้ตบุ๊กของคุณจะใช้ GPU ฟรีที่มีให้ในระบบคลาวด์ระหว่างการประมวลผล หากต้องการรับความรู้สึกของการประมวลผล GPU ให้ลองเรียกใช้แอปพลิเคชันตัวอย่างจากMNIST แบบฝึกหัดที่คุณโคลนไว้ก่อนหน้านี้

!python3 "/content/drive/My Drive/app/mnist_cnn.py"

ลองเรียกใช้ไฟล์ Python เดียวกันโดยไม่เปิดใช้งาน GPU คุณสังเกตเห็นความแตกต่างของความเร็วในการดำเนินการหรือไม่?

การทดสอบ GPU

คุณสามารถตรวจสอบได้อย่างง่ายดายว่า GPU เปิดใช้งานหรือไม่โดยเรียกใช้รหัสต่อไปนี้ -

import tensorflow as tf
tf.test.gpu_device_name()

หากเปิดใช้งาน GPU จะให้ผลลัพธ์ดังต่อไปนี้ -

'/device:GPU:0'

รายชื่ออุปกรณ์

หากคุณอยากรู้ว่าอุปกรณ์ที่ใช้ระหว่างการเรียกใช้โน้ตบุ๊กของคุณในระบบคลาวด์ให้ลองใช้รหัสต่อไปนี้ -

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

คุณจะเห็นผลลัพธ์ดังนี้ -

[name: "/device:CPU:0"
   device_type: "CPU"
   memory_limit: 268435456
   locality { }
   incarnation: 1734904979049303143, name: "/device:XLA_CPU:0"
   device_type: "XLA_CPU" memory_limit: 17179869184
   locality { } 
   incarnation: 16069148927281628039
   physical_device_desc: "device: XLA_CPU device", name: "/device:XLA_GPU:0"
   device_type: "XLA_GPU"
   memory_limit: 17179869184
   locality { }
   incarnation: 16623465188569787091
   physical_device_desc: "device: XLA_GPU device", name: "/device:GPU:0"
   device_type: "GPU"
   memory_limit: 14062547764
   locality {
      bus_id: 1
      links { } 
   }
   incarnation: 6674128802944374158
physical_device_desc: "device: 0, name: Tesla T4, pci bus id: 0000:00:04.0, compute capability: 7.5"]

กำลังตรวจสอบ RAM

หากต้องการดูทรัพยากรหน่วยความจำที่พร้อมใช้งานสำหรับกระบวนการของคุณให้พิมพ์คำสั่งต่อไปนี้ -

!cat /proc/meminfo

คุณจะเห็นผลลัพธ์ต่อไปนี้ -

MemTotal: 13335276 kB
MemFree: 7322964 kB
MemAvailable: 10519168 kB
Buffers: 95732 kB
Cached: 2787632 kB
SwapCached: 0 kB
Active: 2433984 kB
Inactive: 3060124 kB
Active(anon): 2101704 kB
Inactive(anon): 22880 kB
Active(file): 332280 kB
Inactive(file): 3037244 kB
Unevictable: 0 kB
Mlocked: 0 kB
SwapTotal: 0 kB
SwapFree: 0 kB
Dirty: 412 kB
Writeback: 0 kB
AnonPages: 2610780 kB
Mapped: 838200 kB
Shmem: 23436 kB
Slab: 183240 kB
SReclaimable: 135324 kB
SUnreclaim: 47916
kBKernelStack: 4992 kB
PageTables: 13600 kB
NFS_Unstable: 0 kB
Bounce: 0 kB
WritebackTmp: 0 kB
CommitLimit: 6667636 kB
Committed_AS: 4801380 kB
VmallocTotal: 34359738367 kB
VmallocUsed: 0 kB
VmallocChunk: 0 kB
AnonHugePages: 0 kB
ShmemHugePages: 0 kB
ShmemPmdMapped: 0 kB
HugePages_Total: 0
HugePages_Free: 0
HugePages_Rsvd: 0
HugePages_Surp: 0
Hugepagesize: 2048 kB
DirectMap4k: 303092 kB
DirectMap2M: 5988352 kB
DirectMap1G: 9437184 kB

ตอนนี้คุณพร้อมสำหรับการพัฒนาโมเดลแมชชีนเลิร์นนิงใน Python โดยใช้ Google Colab