最近想尝试一下机器学习的东西,但是苹果笔记本大多没有独立显卡,不过大家可以搞一个拓展坞,但这个东西对BootCamp支持特别差劲,难怪卖东西的人都说没试过BootCamp,明显是不好用。不过还好在macos下用起来还是比较舒服的。
如果想尝试一下gpu版本的tensorflow,那么不妨按照我的教程走下去,网上那群人我也是醉了,各种奇奇怪怪的说法。
CPU版本就不多说了,直接pip3 install tensorflow就好了。
既然都开始接触机器学习了还想用GPU加速,我想你的Python应该知道安装pip了吧。
首先是安装GPU驱动,去官网下载就好了,然后下载cuda,也是官网下载安装包直接安装就好。之后就是下载cudnn,这个是Nvdia提供的一个机器学习的框架,tensorflow底层用的就是这个框架,所以不安装就没法运行了。cudnn在国内的Nvdia的网站总是在维护,我的博客空间也小,上传不了这么大的文件,麻烦大家自己科技一下,然后下载吧。我现在是2017年9月,这个时候mac上的tensorflow支持的是cudnn5.1for cuda 8.0。
下载下来之后是一个压缩包,解压后一个是lib(64)文件夹一个是include文件夹,
1.sudo cp include/* /Developer/NVIDIA/CUDA-8.0/include/
2.sudo cp lib(64)/* /Developer/NVIDIA/CUDA-8.0/lib/
3.sudo cp include/* /usr/local/cuda/include
4.sudo cp lib/* /usr/local/cuda/lib(其实