1. はじめに PyTorchで、Pythonのパッケージは、condaでインストールする場合が多い。しかし、今回は、pipベースでやってみた。また、Google Colabでも出来るのでそちらについても記載する。 2. パッケージの準備手順 コンパイルに使ったOSは、Ubuntu Server LTS 18.04もしくは16.04 (HVM) (EC2)である。PyTorch 1.0.0以降は、Ubuntu Server 18.04を使っている。さらに、PyTorch 1.5.0以降では、Ubuntu 20.04(x86_64/aarch64)を使っている。なお、18.04の場合、clock_gettimeのRelinkの問題があり動かない。しかし、USE_MKLDNNをdisableとすると動く。 2.1. CUDAライブラリの設定 CUDAライブラリのインストールは以下の通りである。
![PyTorchのビルド方法とビルド時間 - Qiita](https://cdn-ak-scissors.b.st-hatena.com/image/square/94c6f4f79a86de98b649239be0b59f9fbab8fd3d/height=288;version=1;width=512/https%3A%2F%2Fqiita-user-contents.imgix.net%2Fhttps%253A%252F%252Fcdn.qiita.com%252Fassets%252Fpublic%252Farticle-ogp-background-412672c5f0600ab9a64263b751f1bc81.png%3Fixlib%3Drb-4.0.0%26w%3D1200%26mark64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZ3PTk3MiZoPTM3OCZ0eHQ9UHlUb3JjaCVFMyU4MSVBRSVFMyU4MyU5MyVFMyU4MyVBQiVFMyU4MyU4OSVFNiU5NiVCOSVFNiVCMyU5NSVFMyU4MSVBOCVFMyU4MyU5MyVFMyU4MyVBQiVFMyU4MyU4OSVFNiU5OSU4MiVFOSU5NiU5MyZ0eHQtYWxpZ249bGVmdCUyQ3RvcCZ0eHQtY29sb3I9JTIzMjEyMTIxJnR4dC1mb250PUhpcmFnaW5vJTIwU2FucyUyMFc2JnR4dC1zaXplPTU2JnM9NzhiMzAwODBhODE5ZjdlMDYxMzRjZmFhZjBlOTE5Mzg%26mark-x%3D142%26mark-y%3D57%26blend64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZoPTc2Jnc9NzcwJnR4dD0lNDBzYWthaWEmdHh0LWNvbG9yPSUyMzIxMjEyMSZ0eHQtZm9udD1IaXJhZ2lubyUyMFNhbnMlMjBXNiZ0eHQtc2l6ZT0zNiZ0eHQtYWxpZ249bGVmdCUyQ3RvcCZzPWE5MzM0N2Y2ODgyYmE2Y2VhMWIzNTk3YmI2MjRjNTYy%26blend-x%3D142%26blend-y%3D486%26blend-mode%3Dnormal%26s%3D9d30f903f3fe134ff9409a22d359983e)