Interestingly, I'm not getting this error on my home machine. I'm using the same GPUs and the same docker image, so the versions are exactly the same - except the nvidia driver is 525.89.02 instead of 525.78.01.kooshi mentioned this issue Apr 14, 2023 Enabling model parallelism (...