Install Rapids.AI from cloud-init

install.sh

./packet device create -f dfw2 -o ubuntu_18_04 -P x2.xlarge.x86 -H rapidsai -p $PACKETPROJECT -u "$(curl https://gist.githubusercontent.com/vielmetti/6b8e689a73ead8958b3f6dadd0b5b920/raw/8c7714b8bf8cf18b21055ea93f1e7adc90b426e9/rapids-ai-docker-cloudinit.sh)"

rapids-ai-docker-cloudinit.sh

#!/bin/sh
export DEBIAN_FRONTEND=noninteractive

apt-add-repository ppa:ansible/ansible
apt-get update && apt-get upgrade -y
apt-get install pciutils docker.io linux-headers-$(uname -r) ansible -y

ansible-galaxy install nvidia.nvidia_driver nvidia.nvidia_docker andrewrothstein.anaconda

cat > /root/nvidia-driver.yml <<EOF
---
- hosts: localhost
  roles: 
   - nvidia.nvidia_driver
EOF

cat > /root/nvidia-docker+anaconda.yml <<EOF2
---
- hosts: localhost
  roles: 
   - nvidia.nvidia_docker
   - andrewrothstein.anaconda
EOF2

ansible-playbook /root/nvidia-driver.yml
ansible-playbook /root/nvidia-docker+anaconda.yml


# /usr/local/anaconda/bin/conda install -y -c rapidsai -c nvidia -c conda-forge -c defaults rapids=0.15 python=3.7 cudatoolkit=11.0

/usr/local/anaconda/bin/conda install -y -c conda-forge mamba
/usr/local/anaconda/bin/mamba install -y -c rapidsai -c nvidia -c conda-forge -c defaults rapids=0.15 python=3.7 cudatoolkit=11.0

shutdown -r now