Try nvidia-docker. It's pretty awesome!