Caffe-TSN 多GPU并行运算OpenMPI

TSN算是Action Recognition比较经典的方法,之前在刚入门的时候跑过一遍这个代码,当时只用了一块GPU来跑,并没有关注该版本的caffe是使用了openMPI来实现并行运算的.貌似从没有关注过多GPU训练,因为只有一张卡啊~ 一开始以为按照通用的设置caffe中多GPU的使用方法,设置完成以后,总是提示数值无效的错误. 然后我又看到解决方案里面有人说需要打开nccl编译的选项,ca
相关文章
相关标签/搜索