近期,我在电脑上安装了tensorflow gpu版本,遇到一些问题,在这里写一些解决方案。
1、安装python3
2、安装tensorflow gpu:pip install tensorflow-gpu
3、查看显示适配器中关于NVIDIA,https://developer.nvidia.com/cuda-gpus,查看是否支持cuda
4、下载cuda,https://developer.nvidia.com/cuda-downloads,根据我的安装,发现tensorflow只支持cuda8.0
5、下载cudnn v5.1,cudnnv6.0安装完后会报错,将cudnn中lib,include,bin中文件移到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 ,可以根据你的路径变换。
6、测试,打开python,输入import tensorflow as tf,如无报错,则安装成功
7、如果报错 No module named '_pywrap_tensorflow_internal',则是因为缺少msvcp140.dll,需要安装visual c++ 2015 redistributable Package,https://www.microsoft.com/en-us/download/details.aspx?id=52685