st_gcn_twostream网络报错 cannot find GPU device

gpu.py第35行调佣了cuda()，但是之前并没有让GPU可见，要在

for g in gus:之前加入

os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(list(map(str, gpus)))

最后要记得，重新在torchlight下安装setup.py，这时会将gpu.py拷贝到python系统安装路径，实际调用的是这里。