在安装gpu版本的tensorflow时,遇到过很多坑。现在把自己电脑的配置及安装过程写出来,跟自己电脑相同配置的人可以做个参考。
1、配置需求:
操作系统:win10 64位
GPU:英伟达GeForce GTX 1080 Ti
cuda9.1版本
cudnn7
python3.6
pip10.0.1
2、安装过程
1)下载安装anaconda
2)安装cuda9.1和cudnn7
3)安装tensorflow-gpu
本人在安装过程中参照的博客为:https://blog.csdn.net/weixin_39290638/article/details/80045236,如果你用的显卡是GPUGTX 1080 Ti型号可以安装我的配置cuda、cudnn、pip相应的版本。当然具体安装过程可以参照上面提到的这篇博客。