I´m trying out PyTorch's DCGAN Tutorial, using a Mac with an M1 chip.
For some reason, when loading images with the Dataloader, the resulting samples are corrupted when using: device = torch.device("mps")
. But when using the device = torch.device("cpu")
I get the correct result as shown below:
I checked that MPS is correctly configured in my environment. I'm using PyTorch's nightly build, which supports MPS acceleration on my MacOs version.
Is this a known issue?