项目地址(Github)
更新2018-03-27
- 增加了对Python 3.6的支持。
- 此博文将不再更新,请关注项目获取更新信息。
Tensorflow OSX Build
很不幸,Tensorflow团队自从1.2版本开始停止了发布 Mac OS CUDA版。本项目提供 Mac OS 上编译好、可直接安装的的Tensorflow CUDA版本。
如果你已经在Mac上安装过Tensorflow-gpu 1.0 / 1.1,可以通过本项目升级到新版本。
版本
FileName | Tensorflow | CUDA | CUDNN | CUDA Capability | Compilation Time |
---|---|---|---|---|---|
tensorflow-1.4.0.dev0-cp27-cp27m-macosx_10_12_intel.whl | 1.4.0 dev0 | 8.0 | 6 | 3.0,3.5,5.2,6.1 | 2017-09-15 |
tensorflow-1.4.0-cp27-cp27m-macosx_10_12_intel.whl | 1.4.0 | 8.0 | 6 | 3.0,3.5,5.2,6.1 | 2017-11-29 |
安装方法
首先,确保CUDA驱动和cudnn正确安装。
其次,卸载之前版本的Tensorflow:
pip uninstall tensorflow
pip uninstall tensorflow-gpu
最后,解压并安装:
cat tensorflow* > tensorflow.whl.zip
unzip tensorflow.whl.zip
pip install tensorflow*.whl
开始使用新版Tensorflow吧!
Tensorflow源代码
Source code from: https://github.com/tensorflow/tensorflow
Pytorch OSX Build
与Tensorflow类似,请直接访问项目主页:
https://github.com/TomHeaven/pytorch-osx-build