檢查是否有可用的 GPU: 使用 torch.cuda.is_available() 函數(shù)來(lái)檢查是否有 GPU 支持 CUDA。

將模型移至 GPU: 使用 .to(device) 方法將模型移至 GPU，其中 device 是一個(gè)字符串，通常設(shè)置為 "cuda:0"（表示第一個(gè) GPU）。

將數(shù)據(jù)移至 GPU: 與模型相同，您還需要確保輸入數(shù)據(jù)和標(biāo)簽也在 GPU 上。在將數(shù)據(jù)送入模型之前，同樣使用 .to(device) 方法。