如果是练习教程中的例子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
安装完成后再偏好设置里会看到这个选项