Dlib未在Google Colab上使用GPU

时间:2020-08-14 10:36:33

标签: python google-colaboratory dlib

如何在GPU上进行强制训练?

当前,即使我运行dlib.DLIB_USE_CUDA并且显示true,它也仅使用CPU。

当我运行1时,它还会显示print(dlib.cuda.get_num_devices())

下面的图片显示了实际上我在运行代码时,GPU上没有任何运行:

enter image description here

注意:GPU设置为RUn

1 个答案:

答案 0 :(得分:1)

评论:

显然,正如我所测试的,这不是训练错误,而是加载错误。加载ibug-300W文件需要花费大量时间和精力。有什么办法可以更快地加载它吗?

如果有人在google colab上偶然发现了此问题或问题(培训时间缓慢)。

更快地加载它的方法是直接在colab的vm / content上传输数据集。因为Drive和Colab之间的传输速度很慢。

PS:加载ibug-300W文件至少需要14-15GB的内存。