Tensorflow和Pytorch Mac版(支持CUDA,Py27、Py36)

项目地址(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,可以通过本项目升级到新版本。

版本

FileNameTensorflowCUDACUDNNCUDA CapabilityCompilation Time
tensorflow-1.4.0.dev0-cp27-cp27m-macosx_10_12_intel.whl1.4.0 dev08.063.0,3.5,5.2,6.12017-09-15
tensorflow-1.4.0-cp27-cp27m-macosx_10_12_intel.whl1.4.08.063.0,3.5,5.2,6.12017-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

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

TomHeaven

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值