GPU资源紧张时PYG NeighborSampler报错原因分析

在NeighborSampler处报错如下。
File "/data/ChenWei/miniconda3/envs/torch/lib/python3.7/site-packages/torch_geometric/data/sampler.py", line 80, in __init__ edge_attr = torch.arange(edge_index.size(1)).cuda() RuntimeError: CUDA error: out of memory
环境是gpu：0的资源已经用尽，使用os设置使用gpu：3，但是总是一到图采样就cuda out of mem。

经过错误排除不是数据问题，知识单方面的函数问题。

所以就仔细分析了一下报错内容，发现sampler代码内.cuda()默认使用了gpu:0，所以导致了cuda out of mem。所以我进入pyg包内将其修改为了.to(torch.device("cuda:3"))就解决了。