RuntimeError: Unexpected error from cudaGetDeviceCount()

torch.package / torch::deploy
14

I don’t know the differences between your local run and the docker one, but the same error was fixed here by reinstalling the NVIDIA driver.