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