Bug with dataparallel in Pytorch 1.7 + cu110

tristandeleu / pytorch-meta

A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch

https://tristandeleu.github.io/pytorch-meta/

MIT License

1.97k stars 256 forks source link

Bug with dataparallel in Pytorch 1.7 + cu110 #153

Open ximinng opened 2 years ago

ximinng commented 2 years ago

Hi, thanks for providing this cool library! But I ran into some problems. The code runs successfully on single GPU. When we apply Dataparallel, it prompts:

Thanks for your help very much!