如何在Mac电脑上面编译GPU版本TensorFlow

2025-05-16 22:46:14
推荐回答(1个)
回答1:

确定你的Mac是Nvidia显卡,且compute capabilities >= 3.0,点这里查看 你的显卡型号是否支持

确保你的显存至少1GB以上(Mac即使是N卡,内存都少得可怜,我的GT640M只有512M,所以后面跑CNN基本都会OOM)

编译TensorFlow需要安装Xcode(如果安装CUDA 7.5.27,可以用Xcode7.3,否者只能用7.2版本)

假定大家的Mac已经安装了Homebrew(没安装的人点这里安装)

在后续安装过程中有些包下载会非常慢,甚至被墙,所以你最好有VPN,如果不想花钱可以用免费的Lantern