I don’t quite understand why this error is raised in the posted code, as it’s pointing to a deserialization in a non-CUDA installation.
Based on the code snippet I assume you might be using CUDA tensors? If so, could you push them to the CPU before trying to store them via pandas?
1 Like