Oh, right... the Docker image running on the instance takes care of the library versions. You guys are great!
I would look for an AMI that has already cuda and all drivers installed. Same goes for the docker image
Make sure it also has docker installed 🙂