Pytorch DataLoader不能与远程解释器一起工作

2024-04-25 03:48:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我有以下错误

Expected: /home/ubuntu/.pycharm_helpers/pydev/pydevd_attach_to_process/attach_linux_amd64.so to exist.

代码如下:

import torch_geometric.transforms as T
category = 'Airplane'
path = osp.join(osp.dirname(osp.realpath(__file__)), '..', 'data', 'ShapeNet')
transform = T.Compose([
    T.RandomTranslate(0.01),
    T.RandomRotate(15, axis=0),
    T.RandomRotate(15, axis=1),
    T.RandomRotate(15, axis=2)
])
pre_transform = T.NormalizeScale()
train_dataset = ShapeNet(path, category, split='trainval', transform=transform,
                         pre_transform=pre_transform)
test_dataset = ShapeNet(path, category, split='test',
                        pre_transform=pre_transform)
train_loader = DataLoader(train_dataset, batch_size=12, shuffle=True, num_workers=6)
test_loader = DataLoader(test_dataset, batch_size=12, shuffle=False,
                         num_workers=6)

当我尝试使用dataloader从数据集采样时,调试器崩溃并返回此错误

已尝试删除远程帮助程序,但没有解决我的问题。 我的本地机器运行在windows 10上,远程机器运行在Ubuntu 18.04上


Tags: topathtest错误transformtrainpredataset
1条回答
网友
1楼 · 发布于 2024-04-25 03:48:08

在这里回答我自己

我试着遵循这个:https://intellij-support.jetbrains.com/hc/en-us/community/posts/360006791159 pycharm-helpers-pydev-pydevd-attach-to-process-attach-linux-amd64-so-undefined-symbol-AttachDebuggerTracing

但事实证明它效率低下

相反,从数据加载器中删除参数'num_workers'解决了这个问题。所以这是一个简单的解决方法

然而,我还没有意识到这个问题。可能是多处理有问题。但是如果有人知道发生了什么。我很乐意接受

相关问题 更多 >