美文网首页Python 各类包环境配置 及 使用技巧
Tensorflow-gpu版搭建时需要的配置

Tensorflow-gpu版搭建时需要的配置

作者: 廿怎么念 | 来源:发表于2020-06-13 10:03 被阅读0次

目前(2020/6/13), Tensorflow-gpu版安装时,需要对应的python、CUDA 和cuDNN版本如下:


Windows系统

查看更新、其他系统、tensorflow cpu版等相关信息,点击此处 另外,如果出现Tensorflow runtime error,那可能是没有安装正确版本的Microsoft Visual C++ Redistributable包,按照图中Complier对应的版本安装即可。

按照上面的信息测试了tensorflow-gpu1.2.0的安装,没有问题。
另外tensorflow-gpu2.2.0 的cuDNN用了7.6.5 搭配cuda10.1 (NVIDIA推荐的),也没有问题,附图如下:


cuDNN与CUDA搭配

相关链接
CUDA Toolkit Archive
cuDNN download

相关文章

网友评论

    本文标题:Tensorflow-gpu版搭建时需要的配置

    本文链接:https://www.haomeiwen.com/subject/hvkatktx.html