https://blog.alicey.dev/tags/ml/Recent content in ML on Alicey's blogHugojaMon, 28 Mar 2022 20:50:15 +0900https://blog.alicey.dev/2021/08/ubuntu2004dockergpu.htmlTue, 03 Aug 2021 10:01:00 +0900https://blog.alicey.dev/2021/08/ubuntu2004dockergpu.html<h2> なりゆき</h2><div>大学で行っている研究にて、機械学習を用いた処理を行っているのですが、学生の持っているマシン(大半がm3なMacBook)では遅すぎていつまで経っても終わらないという話が出ました。そこで、大学に機械学習のリソースをくれないかという話をしたところ、GPU搭載のゲーミングノートPCを貸していただけたので、これを使うことにしました。</div><div>（本当はGPU乗ったサーバのVM貸してほしかった.....電気代怖い....）</div><div><br></div><div>PCの詳細はこんな感じ</div><div><ul><li>CPU : Intel Core i7 9750H(6C12T 2.6GHz up to 4.5GHz)</li><li>RAM : DDR4-2400 SODIMM 32GB</li><li>GPU : NVIDIA RTX2070 Mobile</li><li>OS : Windows10Home → Ubuntu 20.04 LTS</li></ul></div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/1a6dca740247810f_hu3304391076628058705.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/1a6dca740247810f_hu3304391076628058705.webp" alt="" width="1293" height="920" loading="lazy" decoding="async"> </a><br><br></div><div><br></div><h2>OSの載せ替え</h2><div>まず、既存のWindowsをバックアップします。</div><div>SSDを載せ替えても良かったのですが、予算の関係で今回はディスクのイメージを作成しました。</div><div>詳しい内容は<a href="https://blog.alicey.dev/2021/07/windowsdd.html" target="_blank">前回の記事</a>を御覧ください。</div><div><br></div><h2>Ubuntuのインストール</h2><div>Ubuntuをインストールします。</div><div>といっても、難しいことはなく、USBブートし画面の指示に従いインストールするだけです。</div><div><br></div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/c26ec61a3e653565_hu14426114048247843325.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/c26ec61a3e653565_hu14426114048247843325.webp" alt="" width="1064" height="797" loading="lazy" decoding="async"> </a><br><br></div><h2>ドライバのインストール</h2><div>ドライバ類のインストールをします。</div><div>今回は、GPUのドライバのみ手動インストールしました。</div><div>バージョンに関しては、<a href="https://www.tensorflow.org/install/gpu?hl=ja" rel="nofollow" target="_blank">TensoeflowのGPUサポート</a>と照らし合わせ、</div><div>CUDA 11.4, NVIDIA-SMI 470.42.01をインストールしました。</div><div><br></div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/40b50e33b212a9e8_hu11333822182802677762.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/40b50e33b212a9e8_hu11333822182802677762.webp" alt="" width="1181" height="827" loading="lazy" decoding="async"> </a><br>ココで1箇所ハマったことがあります。</div><div>UbuntuをSecureBootを利用する形でセットアップしたため、ドライバインストール時に見慣れない「Enrolling MOK Key」の設定が必要になります。</div><div>ドライバをセットアップし、Reboot後、見慣れない青背景の画面が出たため、何も考えずにContinueBootした所、ドライバを読み込みませんでした。気をつけましょう。</div><div><br></div><div>その時参考にしたサイト: <a href="https://seedoftalk.blogspot.com/2021/06/ubuntu-secureboot.html" rel="nofollow" target="_blank">UbuntuでNVIDIAドライバが動かない</a></div><div><br></div><h2>Docker等の構築</h2><div>ココまで来てしまえば、後は必要なソフトウェアをインストールするだけです。</div><div>Tensorflowの場合ですと、公式に<a href="https://www.tensorflow.org/install/docker?hl=ja" rel="nofollow" target="_blank">Docker</a>を利用するものと、<a href="https://www.tensorflow.org/install/pip?hl=ja" rel="nofollow" target="_blank">pip</a>でのセットアップに関するものがまとめられています。</div><div>これを参考に構築します。</div><div>（需要があれば、別途記事にします。）</div><div><br></div><div>また、DockerでNVIDIA GPUを利用する際にはNVIDIA Container Toolkit(nvidia-docker2)のインストールが必要になります。</div><div><a href="https://gangannikki.hatenadiary.jp/entry/2020/09/26/200000" rel="nofollow" target="_blank">こちら</a>を参考にさせていただきました。</div><div>公式の<a href="https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#setting-up-nvidia-container-toolkit" rel="nofollow" target="_blank">ドキュメント</a>はこの辺りですね。</div><div><br></div><div>docker composeを利用する場合、v1とv2でdocker-compose.ymlのGPUを指定する方法が異なりますのでご注意ください。個人的にはv2で構築することをおすすめします。</div><div>詳しくは、<a href="https://matsuand.github.io/docs.docker.jp.onthefly/compose/gpu-support/" rel="nofollow" target="_blank">ドキュメント</a>をご参照ください。</div><div><br></div><div>また、この辺りは、バージョンによって変化が生じやすい部分かと思われますので、環境(OSやGPU)に応じた設定を都度ドキュメント等で確認することをおすすめします。</div><div><br></div><div><br></div><h2>おまけ</h2><div>せっかくゲーミングPCだからゲームしないとね？</div><div>というわけで、マイクラをプレイしておきました。</div><div><br></div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/caba0afd9f74752f_hu16241844205032627848.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/caba0afd9f74752f_hu16241844205032627848.webp" alt="" width="726" height="920" loading="lazy" decoding="async"> </a><div><br></div><div>僕がよくする配信+ゲームではGPUフルロード。フレームレートも落ちますね....144Hzモニタの意味....</div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/82426cc5ab1153e6_hu15285195861590897201.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/82426cc5ab1153e6_hu15285195861590897201.webp" alt="" width="1031" height="730" loading="lazy" decoding="async"> </a><br><br></div><br><br></div><div>というか、キー配列無茶苦茶すぎん？</div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/abaf06300e8ef4a4_hu3461504936845575084.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/abaf06300e8ef4a4_hu3461504936845575084.webp" alt="" width="1600" height="987" loading="lazy" decoding="async"> </a><br><br></div><div>無駄にBIOSからキーボードの光り方変えられるｗ</div><div><a href="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/a17ea08e664c9044_hu6141220186596196352.webp" class="lightbox-trigger"> <img src="https://blog.alicey.dev/2021/08/ubuntu2004dockergpu/img/a17ea08e664c9044_hu6141220186596196352.webp" alt="" width="1260" height="900" loading="lazy" decoding="async"> </a><div><br></div><br>いろいろと癖が強いマシンだった。</div><div><br></div><div><br></div><div><br></div><div><br></div><div><br></div><div><br></div><div><br></div><div>ばいち</div>