Geo-CNN-Pytorch-PYG icon indicating copy to clipboard operation
Geo-CNN-Pytorch-PYG copied to clipboard

Published 20 hours ago • verified publisher icon cy69855522

GPU problem

Open zz-cloud opened this issue 3 years ago • 2 comments

  • you run the codes on a single gpu or multi gpu? some differences exist in the code? thanks!

zz-cloud avatar May 05 '21 00:05 zz-cloud

I run the code on a single GPU, but it supports multiple GPU mode. It's implemented by dp of PyTorch, btw, ddp may be better.

cy69855522 avatar May 05 '21 05:05 cy69855522

How to use DataListloader of PyG to transfer pin_memory TRUE to dataloader of PyTorch? I want to load data to GPU.

zz-cloud avatar May 17 '21 01:05 zz-cloud