1. 程式人生 > 其它 >pytorch 提高GPU利用率 TensorFlow可參考

pytorch 提高GPU利用率 TensorFlow可參考

技術標籤:pytorch錯誤總結pytorchtensorflow深度學習

torch.utils.data.DataLoader(image_datasets[x],
                            batch_size=batch_size, 
                            shuffle=True,
                            num_workers=8,
                            pin_memory=True)
  1. 將num_workers調大,儘量大,多程序呼叫
  2. 設定pin_memory為 True ,由CPU傳輸的資料不需要經過RAM,直接對映到GPU上
  3. batch_size 儘量大