- 博客(1)
- 资源 (1)
- 收藏
- 关注
原创 keras CUDNN_STATUS_BAD_PARAM core dump 问题
使用keras训练模型时遇到 CUDNN_STATUS_BAD_PARAM core dump 问题,发现时因为我使用的是GPU并行方式来训练模型,最后一个batch的样本如果没有大于等于GPU的个数的话,就会发生这个问题。我解决的方法是,设置合适的batch_size,让最后一个batch的样本个数大于等于GPU的个数。因为并行模式里,每个GPU至少要有一个样本来跑。...
2018-08-16 21:11:40
937
1
xgboost 0.8.1 linux 多GPU版本
官网下的 xgboost 0.8.1 linux 多GPU版本,安装后发现只有单支持个GPU, 自己编译了多GPU的 libxgboost.so ,基于cuda 9.0的, 安装后替换 ~/anaconda/xgboost (需要改成你自己的路径) 里的libxgboost.so 就可以了
2019-01-15
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人