Caffe-TSN 多GPU並行運算OpenMPI

TSN算是Action Recognition比較經典的方法,之前在剛入門的時候跑過一遍這個代碼,當時只用了一塊GPU來跑,並沒有關注該版本的caffe是使用了openMPI來實現並行運算的.貌似從沒有關注過多GPU訓練,因爲只有一張卡啊~ 一開始以爲按照通用的設置caffe中多GPU的使用方法,設置完成以後,總是提示數值無效的錯誤. 然後我又看到解決方案裏面有人說需要打開nccl編譯的選項,ca

>>阅读原文<<