如何解决如果我在 Cuda 上跟踪 PyTorch 网络，我可以在 CPU 上使用它吗？
我在兼容 CUDA 的 GPU 服务器上使用 torch.jit.trace 跟踪了我的神经网络。当我在同一台服务器上重新加载该 Trace 时，我可以重新加载它并正常使用它。现在，当我将它下载到我的笔记本电脑上（用于快速测试）时，当我尝试加载跟踪时，我得到：
RuntimeError: Could not run 'aten::empty_strided' with arguments from the 'CUDA' backend. 'aten::empty_strided' is only available for these backends: [cpu,BackendSelect,Named,AutogradOther,Autogradcpu,AutogradCUDA,AutogradXLA,AutogradPrivateUse1,AutogradPrivateUse2,AutogradPrivateUse3,Tracer,Autocast,Batched,VmapMode].
我不能在跟踪中在 GPU 和 cpu 之间切换吗？还是有其他事情发生？
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
