[Tips] PyTorchにおける動的リンク

はじめに Turingの基盤AIチームに業務委託として所属している東京科学大学(Institute of Science Tokyo)の藤井です。 本記事では、PyTorchをsource buildした後に利用する際にCUDA Toolkitをmodule loadしておく必要があることの背景について解説を行います。 また、pip installによりbuild済みのbinaryをインストールしたときのPyTorchとsource buildした場合の違いについても解説を行います。 普段はSwallow Projectや横田研究室にて大規模モデルの分散並列学習や低精度学習について研究を行っていますので、そちらもご覧いただけますと幸いです。 本記事の概要を図にすると以下の通りになります。 pip install時の挙動 まず、source buildしたときとの比較のために、多くの方がよく

[misshiki]

“pip installによりbinary形式のbuild済みPyTorchを利用するときと、source buildしたPyTorchを利用するときに裏で行われている動的リンクに焦点を当てて解説”

[agricola]

まるっとspackで面倒見てもらいたいところなんだが、アレのパッケージは往々にしてパッケージ作成者の俺様環境でしか正常にインストールできない問題が……エラーが出たらレッツapt-get！←泣ける