Traceback (most recent call last):


 "train.py", line 27, in <module>

model.optimize_parameters()


 "/home/wuchen/Downloads/pytorch-CycleGAN-and-pix2pix/models/cycle_gan_mod

self.backward_G()


 "/home/wuchen/Downloads/pytorch-CycleGAN-and-pix2pix/models/cycle_gan_mod

rec_B = self.netG_A(fake_A)


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu

result = self.forward(*input, **kwargs)


 "/home/wuchen/Downloads/pytorch-CycleGAN-and-pix2pix/models/networks.py",

return nn.parallel.data_parallel(self.model, input, self.gpu_ids)


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/parallel/dat

return module(*inputs[0], **module_kwargs[0])


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu

result = self.forward(*input, **kwargs)


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/cont

input = module(input)


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu

result = self.forward(*input, **kwargs)


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv

self.padding, self.dilation, self.groups)


 "/home/wuchen/anaconda2/lib/python2.7/site-packages/torch/nn/functional.p

return f(input, weight, bias)
RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytcu:58
Traceback (most recent call last):


 "./examples/run_squad.py", line 527, in <module>

main()


 "./examples/run_squad.py", line 437, in main

torch.cuda.set_device(args.local_rank)


 "/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py", line 265, in set_device

torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (10) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:33
