为了学caffe搞了块GTX950,勉强满足了深度学习的基本要求
编译好caffe后,开始训练模型,用手机拍了一小段视频,总共500帧图像
训练过程参考了这篇博客
生成模型后训练过程中报错,显示Caffe | Check failed: error == cudaSuccess (2 vs. 0) out of memory
搜索后发现是显存不足,但是我把batch_size改为1之后还是小,又用nvidia-smi命令查看GPU进程,一切都正常
为了学caffe搞了块GTX950,勉强满足了深度学习的基本要求
编译好caffe后,开始训练模型,用手机拍了一小段视频,总共500帧图像
训练过程参考了这篇博客
生成模型后训练过程中报错,显示Caffe | Check failed: error == cudaSuccess (2 vs. 0) out of memory
搜索后发现是显存不足,但是我把batch_size改为1之后还是小,又用nvidia-smi命令查看GPU进程,一切都正常