NVIDIA/DIGITS on EC2

I’m using the NVIDIA/DIGITS6 image on AWS, which works initially but once the machine has been stopped and started again digits fails to launch.

sudo nvidia-docker run --name digits -d -p 5000:5000 -v /home/digits/data:/data nvidia/digits:6.0
nvidia-docker | 2019/07/04 10:50:17 Error: Could not load UVM kernel module. Is nvidia-modprobe installed?

Looks to me like the driver has somehow been lost??? any advice???