如何在 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
提交
取消