3 ms·

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