3 ms·

I've setup pytorch with CUDA on nix and it was quite straightforward, but never tried anything more exotic.