Thank you for this amazing library.
Do you know a simple way to change the GPU device of
param_f = lambda: param.image(self.image_shape[0], self.image_shape[1], batch=1, channels=channels)
In the file lucent.optvis.param.spatial.py the device is set by
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
But I'd like to set the device to "cuda:1" for instance.
Hi,
Thank you for this amazing library. Do you know a simple way to change the GPU device of
param_f = lambda: param.image(self.image_shape[0], self.image_shape[1], batch=1, channels=channels)In the file lucent.optvis.param.spatial.py the device is set by
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")But I'd like to set the device to "cuda:1" for instance.Thanks!