pytorch在MacOS上使用GPU

pytorch在MacOS上使用GPU

安装

直接在pytorch官网按照自己的系统和环境使用命令行安装即可，我这里是M2的MacBook Pro。

使用GPU

pytorch在mac下默认使用的是cpu，因为mac没有cuda，所以需要安装苹果自己的metal库来支持深度学习框架使用GPU，我是在配置TensorFlow的时候配置的metal库，之后安装pytorch之后没有做其他的配置，通过语句选择后端库。

核心代码：

torch.device("mps")

"mps"就是苹果用来实现GPU加速的深度学习后端库。