ansible_user_data_ubuntu

#cloud-config

package_upgrade: true

packages:
 - wget
 - tar
 - git
 - ansible

runcmd:
 - [ modprobe, -r, nouveau ]
 - [ sed, -i, -e, 's/quiet/quiet rdblacklist=nouveau/g', /etc/default/grub ]
 - echo 'blacklist nouveau' >/etc/modprobe.d/disable-nouveau.conf
 - [ grub2-mkconfig, -o, /boot/grub2/grub.cfg ]
 - [ ansible-galaxy, install, LIP-Computing.ansible-role-nvidia ]

final_message: "The system is finally up, after $UPTIME seconds"

users:
  - name: ubuntu
    sudo: ALL=(ALL) NOPASSWD:ALL
    lock-passwd: true
    ssh-import-id: ubuntu
    ssh-authorized-keys:
      - ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAuNBEiH455kxaEr8tWEjdqICI4PDnOus7hTx0Qvix05pySKQuQlSoALk8UsPSHSgFZj9ATpgFFSepACn0hnHu9DISGHn6aIWE8snUc5UctWMlzNoutTngooSjYQxRPZTfHcntjgd5T0Amfqgn1PjO8qaseKLmsR87FhG7Pfe6yodnxqFXu2rzhFYbzkB9MdkFGplCjCT+8+YaGyThxEvB33YHP5hNzCtdqrJ/ojGUgPRZ9QwxrJLgwMOe6EKrngJsVQUfF82CvpQltH82mPDEWCBtLUOx+Mo/PFrvcDMZbgbIEIp9kOPeRLD3jRGObYuANJYaVoxir9N9N2TFbbcDHQ== verlato@prod-ui-02.pd.infn.it

write_files:
  - path: /root/install-gpu-driver.sh
    permissions: 0744
    content: |
      ansible-playbook /etc/ansible/roles/LIP-Computing.ansible-role-nvidia/tests/install-vm.yml --extra-vars nvidia_driver_version=375.39
      nvidia-smi -pm 1
      nvidia-smi -am 1

  - path: /root/install-disvis.sh
    permissions: 0744
    content: |
      curl -fsSL https://get.docker.com/ | sh
      systemctl start docker
      version=$(nvidia-smi | awk '/Driver Version/ {print $6}')
      # Testing DisVis (~100s on Tesla K20m)
      #export WDIR=/usr/local/disvis/test-cases/PRE5-PUP2-complex
      #export ang=10.0
      #export vox=2
      #./udocker run --hostenv dis375 disvis $WDIR/O14250.pdb $WDIR/Q9UT97.pdb $WDIR/restraints.dat -g -a $ang -vs $vox -d $WDIR/example-results
      # nvidia-smi --query-accounted-apps=pid,gpu_serial,gpu_name,gpu_utilization,time --format=csv
      docker run --name=dis1 --device=/dev/nvidia0:/dev/nvidia0 --device=/dev/nvidia1:/dev/nvidia1 --device=/dev/nvidiactl:/dev/nvidiactl --device=/dev/nvidia-uvm:/dev/nvidia-uvm -v /root:/home indigodatacloudapps/disvis:nvdrv_$version /bin/sh -c 'export WDIR=/usr/local/disvis/test-cases/PRE5-PUP2-complex; disvis $WDIR/O14250.pdb $WDIR/Q9UT97.pdb $WDIR/restraints.dat -g -a 10.0 -vs 2 -d /home/example-results'
      nvidia-smi --query-accounted-apps=pid,gpu_serial,gpu_name,gpu_utilization,time --format=csv

  - path: /root/install-powerfit.sh
    permissions: 0744
    content: |
      sudo curl -fsSL https://get.docker.com/ | sh
      sudo systemctl start docker
      version=$(nvidia-smi | awk '/Driver Version/ {print $6}')
      #git clone https://github.com/indigo-dc/udocker
      #cd udocker
      #./udocker pull indigodatacloudapps/powerfit:nvdrv_$version
      #./udocker create --name=pow  indigodatacloudapps/powerfit:nvdrv_$version
      # Testing PowerFit (~5m on Tesla K20m)
      wget http://www.pd.infn.it/~verlato/GPU/appl-files.tgz
      tar zxvf appl-files.tgz
      # export MAP=/home/centos/GroEL-GroES/1046.map
      # export RESOL=46.0
      # export ANG=4.71
      # export PDB=/home/centos/GroEL-GroES/GroES_1gru.pdb
      # export TYPE="-g"
      # export OUT_DIR=/home/centos/outdir
      # udocker/udocker run --hostenv --bindhome pow powerfit ${MAP} ${RESOL} ${PDB} -a $ANG -l ${TYPE} -d ${OUT_DIR}
      time docker run --name=pow1 --device=/dev/nvidia0:/dev/nvidia0 --device=/dev/nvidia1:/dev/nvidia1 --device=/dev/nvidiactl:/dev/nvidiactl --device=/dev/nvidia-uvm:/dev/nvidia-uvm -v /root:/home indigodatacloudapps/powerfit:nvdrv_$version /bin/sh -c 'powerfit /home/GroEL-GroES/1046.map 46.0 /home/GroEL-GroES/GroES_1gru.pdb -a 4.71 -l -g -d /home/out'
      nvidia-smi --query-accounted-apps=pid,gpu_serial,gpu_name,gpu_utilization,time --format=csv

power_state:
 mode: reboot
 delay: "+1"
 message: Bye Bye
 timeout: 120
 condition: True