配置tensorflow GPU 版本填坑路

如果是练习教程中的例子tensorflow cpu 版本够用了,要训练的话还是gpu版本要快很多,

本文记录了在我们配备的主流Mac电脑上,安装gpu版本常见问题和解决方法

显卡为:

  芯片组型号:    NVIDIA GeForce GTX 775M
  类型:    GPU
  总线:    PCIe
  PCIe Lane 宽度:    x16
  VRAM(总和):    2048 MB
  供应商:    NVIDIA (0x10de)  
一. 环境


Mac OSX 10.12
Python:3.6.2
CUDA Toolkit 8.0
cuDNN 5.1

 二. tensorflow 的安装

 参考官方教程,https://www.tensorflow.org/install/install_mac, 不再赘述

建议

1 如果你使用Pycharm编辑器的话,Pycharm自带虚拟环境创建,建议安装到PyCharm的虚拟环境中。

2如果你裸写,建议使用virtualenv来安装tensorflow

三. 安装CUDA

1.安装 CUDA Driver 8.0:

请到如下地址安装最新版 CUDA Driver for Mac:
http://www.nvidia.com/object/mac-driver- archive.html

安装完成后再偏好设置里会看到这个选项

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值