pytorch在MacOS上使用GPU
安装
直接在pytorch官网按照自己的系统和环境使用命令行安装即可,我这里是M2的MacBook Pro。
使用GPU
pytorch在mac下默认使用的是cpu,因为mac没有cuda,所以需要安装苹果自己的metal库来支持深度学习框架使用GPU,我是在配置TensorFlow的时候配置的metal库,之后安装pytorch之后没有做其他的配置,通过语句选择后端库。
核心代码:
torch.device("mps")
"mps"就是苹果用来实现GPU加速的深度学习后端库。