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