安装Tensorflow gpu版本还是比较繁琐的。经常会碰到各种问题。这个是简单的步骤总结:
- 安装anaconda
此时要选好的是anacond和python的版本。 - 创建环境
conda create -n tf - 安装Tensorflow gpu
conda install tensorflow-gpu
此命令与pip install tensorflow-gpu 的区别是安装了多种依赖包,不用去单独安装了。强烈建议使用conda install tenforflow-gpu
一般而言,安装之后,都会碰到gpu不工作的问题。常见的问题如下:
- h5版本问题
此问题的解决方法是重新安装h5:
pip uninstall h5py
pip install h5py - cudatoolkit与显卡驱动不匹配问题
此问题的解决方法是升级显卡驱动。这种的问题产生的一个原因就是电脑是以前生产的或者设计定型的。显卡的驱动不是最新版。nvidia的显卡都是支持tensorflow gpu的最新版本的。出现问题的原因就是显卡驱动是旧版本。
安装显卡驱动最好的方法是使用手动搜索去找安装包。步骤见我的上一篇微博。 - 下载速度慢
建议使用国内的镜像网站