pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --ignore-installed --upgrade tensorflow-gpu==1.8
然后安装cuda9.0,去官网下win10对应的cuda9.0,装好,在添加环境变量,cuda9.0我电脑是默认装在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0这个目录下。而且会给你自动添加两个环境变量:
我们再添加三个环境变量到系统变量path里
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
再确定,确定。这个时候如果在控制台导入tensorflow可能会报错,先退出控制台,在重进控制台,应该可以了
3,把cudnn文件夹目录下的几个文件拷到CUDA对应目录下。怎么下载cudnn,看这篇文章,https://blog.csdn.net/xuefengyang666/article/details/79422012
完成。