Had this same problem again loading a model that I had been able to load before, this fixed the problem:
self.load_state_dict(torch.load(join(exDir, 'params'), map_location=lambda storage, loc: storage.cuda(0)))