用caffe 训练网络时，遇到了Check failed: error == cudaSuccess (2 vs. 0) out of memory
意思是gpu内寸不够用，输入：
nvidia-smi
可以看到最后两个线程占用gpu内存很大。需要使用命令：
sudo kill -9 [线程pid]把不需要的线程杀死，然后就能愉快的训练了
caffe error :Check failed: error == cudaSuccess (2 vs. 0) out of memory
猜你喜欢
转载自blog.csdn.net/qq_34650787/article/details/80494790
今日推荐
周排行