小编Pab*_*lla的帖子

在“mps”设备上使用 pytorch 的数据加载器将无法正确加载图像

我\xc2\xb4m使用带有 M1 芯片的 Mac尝试 PyTorch 的DCGAN 教程。

\n

由于某种原因,当使用数据加载器加载图像时,使用以下命令时生成的样本会损坏:device = torch.device("mps")。但使用时device = torch.device("cpu")我得到正确的结果,如下所示:

\n

结果是正确的

\n

我检查了我的环境中 MPS 的配置是否正确。我正在使用 PyTorch 的 nightly build,它在我的 MacOs 版本上支持 MPS 加速。

\n

这是一个已知的问题?

\n

conda pytorch apple-m1 pytorch-dataloader

5
推荐指数
0
解决办法
1035
查看次数

标签 统计

apple-m1 ×1

conda ×1

pytorch ×1

pytorch-dataloader ×1