ansibleでml用のpythonでgpuを利用できる環境を自動設定する
下記で公開。
https://github.com/notfolder/ml-ansible
下記で環境を構築します。
$ 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