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

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

被折叠的 条评论
为什么被折叠?



