pytorch在MacOS上使用GPU
安装
直接在pytorch官网按照自己的系统和环境使用命令行安装即可，我这里是M2的MacBook Pro。
使用GPU
pytorch在mac下默认使用的是cpu，因为mac没有cuda，所以需要安装苹果自己的metal库来支持深度学习框架使用GPU，我是在配置TensorFlow的时候配置的metal库，之后安装pytorch之后没有做其他的配置，通过语句选择后端库。
核心代码：
torch.device("mps")
"mps"就是苹果用来实现GPU加速的深度学习后端库。
本文转载自: https://blog.csdn.net/Cluluxiu/article/details/141144274
版权归原作者 Cluluxiu 所有， 如有侵权，请联系我们删除。
版权归原作者 Cluluxiu 所有， 如有侵权，请联系我们删除。