You could alternatively use the google AI Platform Notebooks which are basically the same thing and billed per hour. You can now Prebake a docker images with all of the libraries that you need.