如何在 GPU 上強制訓練？目前即使我運行它也只使用 CPU dlib.DLIB_USE_CUDA，它說true。它還說1當我跑步時print(dlib.cuda.get_num_devices())這是附加的圖像，顯示實際上我正在運行代碼時 GPU 上沒有運行任何東西：
1 回答
桃花長相依
TA貢獻1860條經驗 獲得超8個贊
評論：
顯然，正如我所測試的那樣，這不是訓練錯誤，而是加載錯誤。加載 ibug-300W 文件需要很多時間和內存。有什么辦法可以更快地加載它嗎？
如果有人在 google colab 上偶然發現了這個問題或問題（訓練時間慢）。
加載速度更快的方法是直接在 colab 的 vm/content 上傳輸數據集。因為 Drive 和 Colab 之間的傳輸速度很慢。
PS：你需要至少 14-15GB 的內存來加載 ibug-300W 文件。
添加回答
舉報
0/150
提交
取消