Saving my validation notes here in case it helps others run this model, or for if I need to run the model in the future. * Create a `p2.xlarge` instance using the following AMI: `Deep Learning AMI (Amazon Linux 2) Version 27.0` (this container uses Tensorflow but doesn't appear it is configured to use GPU) * `docker login docker.synapse.org` * `docker pull docker.synapse.org/syn21553207/sbnb:9698788` * `mkdir output` * test without training: `docker run -it --rm -v ${PWD}/output:/output --runtime=nvidia docker.synapse.org/syn21553207/sbnb:9698788` @SBNB - Have you tried to run this container with a GPU? Thanks! From what I can tell it only uses CPU.