ansibleでml用のpythonでgpuを利用できる環境を自動設定する


ansibleでml用のpythonでgpuを利用できる環境を自動設定する

下記で公開。
https://github.com/notfolder/ml-ansible

下記で環境を構築します。
terminal
$ ansible-playbook -i investory playbook.yml

ubuntu 18.04を想定し、下記環境を構築します。

  • pyenvをansible_userに設定
  • nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04とほぼ同じセットアップでnvidiaドライバ環境構築
  • nvidia-doker2環境を構築

nvidiaのdokcerイメージと同じ環境を作る方法

下記の様に目的のイメージハッシュのページを開くと、IMAGE LAYERSごとにコマンドを確認できる。
これをansible化しただけ。
https://hub.docker.com/layers/nvidia/cuda/11.1-cudnn8-devel-ubuntu18.04/images/sha256-c5bf5c984998cc18a3f3a741c2bd7187ed860dc6d993b6fb402d0effb9fe6579?context=explore