问题描述
如何在GPU上进行强制训练?
当前,即使我运行dlib.DLIB_USE_CUDA
并且显示true
,它也仅使用cpu。
当我运行1
时,它还会显示print(dlib.cuda.get_num_devices())
下面的图片显示了实际上我在运行代码时,GPU上没有任何运行:
注意:GPU设置为RUn
解决方法
评论:
显然,正如我所测试的,这不是训练错误,而是加载错误。加载ibug-300W文件需要花费大量时间和精力。有什么办法可以更快地加载它吗?
如果有人在google colab上偶然发现了此问题或问题(培训时间缓慢)。
更快地加载它的方法是直接在colab的vm / content上传输数据集。因为Drive和Colab之间的传输速度很慢。
PS:加载ibug-300W文件至少需要14-15GB的内存。