برای استفاده از GPU کافی است مدل و داده را با model.to('cuda') و tensor.to('cuda') به کارت گرافیک منتقل کنی. این کار زمان آموزش را چند برابر کاهش میدهد. قبل از آن با torch.cuda.is_available() بررسی کن GPU فعال است یا نه.
برای مشاهدهٔ ادامه، خرید کنید
دسترسی سریع و فوری