Ubuntu16.04 NCCL1.2.3+Torch+cuda8.0 多GPU加速環境配置

最近在做機器翻譯的實驗,在跑代碼的時候需要用到NVIDIA的NCCL多GPU加速工具,網上有關NCCL的博客都是基於NCCL2.x版本,而我要用到的是1.x版本,經過很多次嘗試終於把環境搞定,在此記錄下我的一些經驗,希望能夠幫助到有需要的小夥伴. 目錄 1. cuda及cudnn安裝 2. Torch安裝 3. NCCL安裝 3.1. 在NCCL2.x環境運行NCCL1.x代碼遇到的問題 3.2
相關文章
相關標籤/搜索