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