Looking into our use, conda helps us install various non-Python cuda libraries that match up with the machine’s driver + PyTorch versions. Looking at the setup in detail it’s not clear to me if that’s needed any more, I’ll come back in a day or so if things don’t work out.