tensorflow GPU版本安装教程 win10

GPU版本安装的时候首先需要安装cuda和cudnn,但是这两个之间有一定的版本关系(很重要)
tensorflow GPU版本安装教程 win10
查看地址为tensorflow官网地址

我在这里下载的cuda和cudnn的版本分别为 (试出来的正确版本),最后使用的tensorflow版本为1.13.1(支持python3.7)

  • cuda_10.0.130_411.31_win10

  • v7.5.0.56

下载完成后进行安装
将cudnn中的文件进行移动
tensorflow GPU版本安装教程 win10

我在这里将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64(\bin,\include,)都添加到了环境变量中,不知道应不应该全加进去

然后安装好之后
tensorflow GPU版本安装教程 win10
打开Geforce进行驱动的更新

我在这里使用的是anaconda环境
接下来使用命令
pip install tensorflow-gpu
安装gpu版本的tensorflow
安装完成后运行代码
中间有一些问题(百度解决了)

参考博客 : 博客一
博客二