Hi ptrblck, thanks for your quick reply. Earlier I was using torch 0.4.0. After getting your reply I updated it into torch 0.4.1 but facing the same problem.
Actually I am running pix2pixHD pytorch model for my image dataset. It run smoothly when I use single GPU but getting error if I want to use multiple GPU. My image size is big and having large dataset so I have to use multiple GPU to save time. Please help me to rid of this problem.