配置tensorflow GPU 版本填坑路

本文详细记录了在Mac电脑上安装Tensorflow GPU版本的过程,包括CUDA和cuDNN的安装,以及遇到的常见问题和解决方法。安装过程中可能会遇到ImportError、Segmentation fault等错误,通过配置环境变量和正确设置CUDA库路径可以解决。同时,文章还提到了Keras与Tensorflow的版本对应关系,确保两者兼容。最后,针对可能出现的OOM错误和显存释放问题,给出了相应的解决建议。

如果是练习教程中的例子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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值