nccl インストールガイド[機械学習でGPUによる学習を並列化したい方向け]

 

nccl, cudnnのインストールが完了したあと

でcupyをインストールします。

nccl, cudnnのインストール、設定が完了する前にcupyをインストールするとちゃんとビルドされないので、
cupyがすでに入ってる場合は一応cupyをアンインストールしてnccl, cudnnを整えてから再度インストールしてください。
このときcacheなしでインストールすることをおすすめします↓。

そして、最後にpythonを開いて、こちらがエラーなく実行できればOKです。

ref: https://qiita.com/pst-ic/items/e01033dee4d389df3a5e