记录一下tensorflow-gpu与cuda版本的对应关系

折腾了好几天终于在新电脑上实现了keras代码自动调用GPU,发现之前失败的主要原因在于各版本对应关系实在太过严苛,故将成功版本记录如下:

系统:win10

python安装方式:Anaconda3 安装的是python3.6.4。

                             具体版本为:Anaconda3-5.1.0-Windows-x86_64.exe

                             下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

                              安装教程:https://mp.csdn.net/postedit/80607252

IDE为:pycharm

接下来是cuda安装:先装VS2015 再装Cuda 最后装cudnn,版本如下:

CUDA版本:cuda_9.0.176_win10.exe

cudnn版本:cudnn-9.0-windows10-x64-v7.3.0.29.zip 

下载地址:https://mp.csdn.net/postedit/85242970

安装教程:借鉴https://jingyan.baidu.com/article/b24f6c821a2f8b86bfe5da19.html

                   https://mp.csdn.net/postedit/86160632

接着如果要执行keras代码,我是以tensorflow作为后端,而不是theano; 为了运行在gpu上,安装tensorflow-gpu,但是不要通过pip install tensorflow-gpu,因为版本一般和CUDA不匹配。

在这里 https://pypi.org/project/tensorflow-gpu/1.12.0/#files下载tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl (80.0 MB)

要不要翻墙我不知道,反正我翻了。用pip install +(将上面下载的whl文件拖进来,回车安装)。

然后同时需要安装tensorflow,在这里 https://pypi.org/project/tensorflow/1.12.0/#files下载tensorflow-1.12.0-cp36-cp36m-win_amd64.whl,用同样的方式安装。

然后安装tensorboard ,在这里 https://pypi.org/project/tensorboard/1.12.0/下载tensorboard-1.12.0-py3-none-any.whl,安装。

然后我的keras就可以自动检测到GPU并运行了,你的如果不行,再找找别的原因,这组版本组合关系应该没啥问题。

猜你喜欢

转载自blog.csdn.net/qq_33485434/article/details/88536548