Upgrade CUDA version in docker images

With the new nvidia-driver 525, a new CUDA version and a newer TensorFlow version should be possible.