Win10安装Tensorflow的Gpu版本问题
作者:
tech0ne | 来源:发表于
2018-08-02 22:16 被阅读0次
Cuda,Cudnn 和 Tensorflow gpu这三个东西要兼容
-
先下载最新的Cudnn,和Cuda的兼容性在下载页上有(需要注册):
Cudnn下载地址
-
下载安装Cudnn的对应Cuda版本,然后将Cudnn解压拷贝到Cuda安装目录:
Cuda下载地址
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\[VER.]
-
上面两个都下载最新相兼容的版本,然后Tensorflow gpu也更新为最新版一般就没问题了:
pip install --upgrade tensorflow-gpu
本文标题:Win10安装Tensorflow的Gpu版本问题
本文链接:https://www.haomeiwen.com/subject/pxkevftx.html
网友评论