训练时报错
RuntimeError: Couldn't open shared file mapping: <torch_16716_3565374679>, error code: <1455>
这可能是因为您的显卡太旧或者计算压力对于您的显卡来说太重了。
就像在 CUDA 张量上进行多线程处理一样无法成功,有两种方法可供选择:
1. 不要使用多线程处理。将 DataLoader 的 num_worker 设置为零。
2. 改为 CPU 共享张量。确保您的自定义 DataSet 数据集返回CPU张量。
方法1有效