tensorflow GPU版本安装教程 win10
GPU版本安装的时候首先需要安装cuda和cudnn,但是这两个之间有一定的版本关系(很重要)
查看地址为tensorflow官网地址
我在这里下载的cuda和cudnn的版本分别为 (试出来的正确版本),最后使用的tensorflow版本为1.13.1(支持python3.7)
-
cuda_10.0.130_411.31_win10
-
v7.5.0.56
下载完成后进行安装
将cudnn中的文件进行移动
我在这里将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64(\bin,\include,)都添加到了环境变量中,不知道应不应该全加进去
然后安装好之后
打开Geforce进行驱动的更新
我在这里使用的是anaconda环境
接下来使用命令
pip install tensorflow-gpu
安装gpu版本的tensorflow
安装完成后运行代码
中间有一些问题(百度解决了)