1.选择合适的cuda和cudnn版本
tensorflow的github:
https://github.com/fo40225/tensorflow-windows-wheel 查看具体适配的版本,
我这里是要安装tf2.0,所以找到相关适配的cuda和cudnn版本:
10.1.243_426.00 / 7.6.4.38
并下载对应的tensorflow_gpu-2.0.0-cp37-cp37m-win_amd64.whl安装包,存入D盘根目录
2.下载相应版本的CUDA 和CUDNN
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/rdp/cudnn-archive
3.安装cuda和cudnn
cuda傻瓜安装,cudnn下载之后是一个压缩包(我这里直接下载到了CUDA的安装目录,因为稍后会解压到这里):
把压缩包里红框的三个文件夹里的东西,
导入到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1”中对应的文件夹;
4.安装tensoflow
进入anaconda命令行,输入“pip install “D:\tensorflow_gpu-2.0.0-cp37-cp37m-win_amd64.whl””
5.测试tensoflow安装是否成功
anaconda命令行,
输入“python”回车,
import tensorflow as tf
tf.__version__
tf.__path__