- cuda drivers
- nvidia container toolkit for GPU support: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
- Make sure to use image with correct version of cuda toolkit / cudnn for version of TF
$ docker build -t overcooked_py37 -f overcooked_py37.Dockerfile .
$ docker run --name overcooked_gpu_instance --gpus all -p 8888:8888 -it overcooked_py37`
$ docker container prune
- setup tailscape
- setup ansible hosts in config