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

目前(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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容