I have a related question, I have a shared model trained on GPU, and another process needs this model for inference on CPU. So I use a shared model and use the following command to load this shared model
    cpu_model.load_state_dict(gpu_model.cpu().state_dict())

however, this won’t work and returns CUDA error(3), initialization error, what happened?