[mlboss]

Try running pytorch/pytorch docker. But you will need nvidia container runtime installed. I am sure somebody will soon release docker for this also.